{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import tensorflow as tf\n",
    "from unicodedata import normalize\n",
    "from tensorflow import keras\n",
    "import os\n",
    "from tensorflow.keras.models import Sequential, load_model, clone_model\n",
    "from tensorflow.keras.optimizers import RMSprop\n",
    "from collections import Counter\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "from matplotlib import pyplot as plt\n",
    "from tensorflow.keras.layers import Conv2D, ZeroPadding2D, AveragePooling2D, BatchNormalization, Activation, Dense, \\\n",
    "    Input, MaxPooling2D,Flatten,Dropout\n",
    "from tensorflow.keras import Model, layers, regularizers\n",
    "import tensorflow.keras.backend as K\n",
    "from tensorflow.keras.datasets import cifar10\n",
    "from tensorflow.keras.callbacks import LearningRateScheduler, ModelCheckpoint, EarlyStopping\n",
    "from tensorflow.keras.preprocessing.image import ImageDataGenerator\n",
    "from tensorflow.keras.initializers import he_normal\n",
    "from tensorflow.keras import optimizers\n",
    "from sklearn.preprocessing import MinMaxScaler\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn import preprocessing\n",
    "from sklearn.metrics import accuracy_score\n",
    "from tqdm import tqdm\n",
    "import csv\n",
    "import cv2\n",
    "#importing model \n",
    "from tensorflow.keras.applications import ResNet50"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "f = csv.reader(open('../../data/data_set/BIT_label.csv','r',encoding='utf-8'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "X_data = []\n",
    "Y_data = []\n",
    "input_shape=(32,32,3)#3通道图像数据\n",
    "for lists in f:\n",
    "    img = cv2.imread('../../data/data_set/BIT/'+lists[1])\n",
    "    img = cv2.resize(img, (input_shape[0], input_shape[1]))\n",
    "    X_data.append(img)\n",
    "    Y_data.append(int(lists[2]))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "num_class = 6"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_x,test_x,train_y,test_y = train_test_split(X_data,Y_data,test_size=0.4)\n",
    "train_x = np.array(train_x).astype('float32') / 255.\n",
    "train_y = np.array(train_y)\n",
    "test_x = np.array(test_x).astype('float32') / 255.\n",
    "test_y = np.array(test_y)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_x_mean = np.mean(train_x, axis=0)\n",
    "test_x_mean = np.mean(test_x, axis=0)\n",
    "train_x -= train_x_mean\n",
    "test_x -= test_x_mean"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "lb = preprocessing.LabelBinarizer().fit(np.array(range(num_class)))  # 对标签进行ont_hot编码\n",
    "train_y = lb.transform(train_y)  # 因为是多分类任务，必须进行编码处理\n",
    "test_y = lb.transform(test_y)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [],
   "source": [
    "def vgg16():\n",
    "    input_tensor = Input(shape=(32, 32, 3))\n",
    "    x = input_tensor\n",
    "    x = Conv2D(64, (3, 3), activation='relu', padding='same', name='block1_conv1')(input_tensor)\n",
    "    x = Conv2D(64, (3, 3), activation='relu', padding='same', name='block1_conv2')(x)\n",
    "    x = MaxPooling2D((2, 2), strides=(2, 2), name='block1_pool')(x)\n",
    "#     print(x.shape)\n",
    "    # Block 2\n",
    "    x = Conv2D(128, (3, 3), activation='relu', padding='same', name='block2_conv1')(x)\n",
    "    x = Conv2D(128, (3, 3), activation='relu', padding='same', name='block2_conv2')(x)\n",
    "    x = MaxPooling2D((2, 2), strides=(2, 2), name='block2_pool')(x)\n",
    "#     print(x.shape)\n",
    "    # Block 3\n",
    "    x = Conv2D(256, (3, 3), activation='relu', padding='same', name='block3_conv1')(x)\n",
    "    x = Conv2D(256, (3, 3), activation='relu', padding='same', name='block3_conv2')(x)\n",
    "    x = Conv2D(256, (3, 3), activation='relu', padding='same', name='block3_conv3')(x)\n",
    "    x = MaxPooling2D((2, 2), strides=(2, 2), name='block3_pool')(x)\n",
    "#     print(x.shape)\n",
    "    # Block 4\n",
    "    x = Conv2D(512, (3, 3), activation='relu', padding='same', name='block4_conv1')(x)\n",
    "    x = Conv2D(512, (3, 3), activation='relu', padding='same', name='block4_conv2')(x)\n",
    "    x = Conv2D(512, (3, 3), activation='relu', padding='same', name='block4_conv3')(x)\n",
    "    x = MaxPooling2D((2, 2), strides=(2, 2), name='block4_pool')(x)\n",
    "#     print(x.shape)\n",
    "    # # Block 5\n",
    "    # x = Conv2D(512, (3, 3), activation='relu', padding='same', name='block5_conv1')(x)\n",
    "    # x = Conv2D(512, (3, 3), activation='relu', padding='same', name='block5_conv2')(x)\n",
    "    # x = Conv2D(512, (3, 3), activation='relu', padding='same', name='block5_conv3')(x)\n",
    "    # x = MaxPooling2D((2, 2), strides=(2, 2), name='block4_pool')(x)\n",
    "    # print(x.shape)\n",
    "\n",
    "    # Classification block\n",
    "    x = Flatten(name='flatten')(x)\n",
    "    x = Dense(4096, activation='relu', name='fc1')(x)\n",
    "    x = Dense(4096, activation='relu', name='fc2')(x)\n",
    "    x = Dropout(0.5)(x)\n",
    "    x = Dense(6, activation='softmax', name='predictions')(x)\n",
    "    model = Model(inputs=input_tensor, outputs=x, name='VGG16')\n",
    "    return model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Model: \"VGG16\"\n",
      "_________________________________________________________________\n",
      "Layer (type)                 Output Shape              Param #   \n",
      "=================================================================\n",
      "input_2 (InputLayer)         [(None, 32, 32, 3)]       0         \n",
      "_________________________________________________________________\n",
      "block1_conv1 (Conv2D)        (None, 32, 32, 64)        1792      \n",
      "_________________________________________________________________\n",
      "block1_conv2 (Conv2D)        (None, 32, 32, 64)        36928     \n",
      "_________________________________________________________________\n",
      "block1_pool (MaxPooling2D)   (None, 16, 16, 64)        0         \n",
      "_________________________________________________________________\n",
      "block2_conv1 (Conv2D)        (None, 16, 16, 128)       73856     \n",
      "_________________________________________________________________\n",
      "block2_conv2 (Conv2D)        (None, 16, 16, 128)       147584    \n",
      "_________________________________________________________________\n",
      "block2_pool (MaxPooling2D)   (None, 8, 8, 128)         0         \n",
      "_________________________________________________________________\n",
      "block3_conv1 (Conv2D)        (None, 8, 8, 256)         295168    \n",
      "_________________________________________________________________\n",
      "block3_conv2 (Conv2D)        (None, 8, 8, 256)         590080    \n",
      "_________________________________________________________________\n",
      "block3_conv3 (Conv2D)        (None, 8, 8, 256)         590080    \n",
      "_________________________________________________________________\n",
      "block3_pool (MaxPooling2D)   (None, 4, 4, 256)         0         \n",
      "_________________________________________________________________\n",
      "block4_conv1 (Conv2D)        (None, 4, 4, 512)         1180160   \n",
      "_________________________________________________________________\n",
      "block4_conv2 (Conv2D)        (None, 4, 4, 512)         2359808   \n",
      "_________________________________________________________________\n",
      "block4_conv3 (Conv2D)        (None, 4, 4, 512)         2359808   \n",
      "_________________________________________________________________\n",
      "block4_pool (MaxPooling2D)   (None, 2, 2, 512)         0         \n",
      "_________________________________________________________________\n",
      "flatten (Flatten)            (None, 2048)              0         \n",
      "_________________________________________________________________\n",
      "fc1 (Dense)                  (None, 4096)              8392704   \n",
      "_________________________________________________________________\n",
      "fc2 (Dense)                  (None, 4096)              16781312  \n",
      "_________________________________________________________________\n",
      "dropout_1 (Dropout)          (None, 4096)              0         \n",
      "_________________________________________________________________\n",
      "predictions (Dense)          (None, 6)                 24582     \n",
      "=================================================================\n",
      "Total params: 32,833,862\n",
      "Trainable params: 32,833,862\n",
      "Non-trainable params: 0\n",
      "_________________________________________________________________\n"
     ]
    }
   ],
   "source": [
    "model = vgg16()\n",
    "model.summary()#显示模型结构"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [],
   "source": [
    "def scheduler(epoch):\n",
    "    if epoch < 80:\n",
    "        return 0.01\n",
    "    if epoch < 160:\n",
    "        return 0.005\n",
    "    return 0.001"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "sgd = optimizers.SGD(lr=0.1, momentum=0.9, nesterov=True)\n",
    "model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['acc'])\n",
    "\n",
    "change_lr = LearningRateScheduler(scheduler)\n",
    "\n",
    "save_dir = os.path.join(os.getcwd(), 'data/trained_model')\n",
    "model_name = 'vgg16_model.{epoch:02d}-{val_acc:.2f}.h5'\n",
    "if not os.path.isdir(save_dir):\n",
    "    os.makedirs(save_dir)\n",
    "filepath = os.path.join(save_dir, model_name)\n",
    "\n",
    "checkpoint = ModelCheckpoint(filepath=filepath,\n",
    "                             monitor='val_acc',\n",
    "                             verbose=1,\n",
    "                             save_best_only=True)\n",
    "\n",
    "#creating early stopping to prevent model from overfitting \n",
    "early_stopping = EarlyStopping(monitor=\"val_acc\", min_delta=0,\n",
    "                                                  patience=30, verbose=1, \n",
    "                                                  mode=\"auto\", baseline=None, \n",
    "                                                  restore_best_weights=True)\n",
    "\n",
    "cbks = [early_stopping, checkpoint, change_lr]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Using real-time data augmentation.\n"
     ]
    }
   ],
   "source": [
    "print('Using real-time data augmentation.')\n",
    "# datagen = ImageDataGenerator(horizontal_flip=True,\n",
    "#                              width_shift_range=0.125, height_shift_range=0.125, fill_mode='constant', cval=0.)\n",
    "datagen = ImageDataGenerator(featurewise_center=True,\n",
    "                                   rotation_range=20,\n",
    "                                   width_shift_range=0.2,\n",
    "                                   height_shift_range=0.2,\n",
    "                                   horizontal_flip=True)\n",
    "valid_data_gen = ImageDataGenerator()\n",
    "datagen.fit(train_x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(5910, 32, 32, 3)"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_x.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(5910, 6)"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_y.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([[[[ 3.49241525e-01,  4.69085276e-01,  4.52662140e-01],\n",
       "         [ 2.56601602e-01,  3.03832561e-01,  3.11898828e-01],\n",
       "         [ 3.31028968e-01,  3.95617038e-01,  3.91022563e-01],\n",
       "         ...,\n",
       "         [ 4.54812050e-02,  6.20465577e-02,  3.58590186e-02],\n",
       "         [ 4.84715104e-02,  6.01733327e-02,  4.68342900e-02],\n",
       "         [ 7.59211481e-02,  8.95334780e-02,  5.97391427e-02]],\n",
       "\n",
       "        [[-3.34545672e-02,  5.61916828e-02,  7.59676695e-02],\n",
       "         [ 1.90554023e-01,  2.93065697e-01,  3.07733387e-01],\n",
       "         [ 3.55216682e-01,  4.39344347e-01,  4.28611249e-01],\n",
       "         ...,\n",
       "         [-1.87111497e-02,  1.22347474e-02,  3.48265767e-02],\n",
       "         [ 5.71253896e-02,  6.51723444e-02,  6.00185692e-02],\n",
       "         [ 2.01810002e-02,  4.68114018e-03,  1.13716722e-03]],\n",
       "\n",
       "        [[ 3.94487798e-01,  4.58996207e-01,  5.03587425e-01],\n",
       "         [ 3.57991844e-01,  4.33477610e-01,  4.30413455e-01],\n",
       "         [ 3.87500405e-01,  3.11542481e-01,  1.06146634e-01],\n",
       "         ...,\n",
       "         [ 5.71262836e-03,  2.50717998e-02,  5.52858710e-02],\n",
       "         [-5.82817793e-02, -3.66979837e-02,  2.33879685e-02],\n",
       "         [ 4.79020476e-02,  5.96216321e-02,  3.26831043e-02]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 8.09918344e-02,  1.07834339e-01,  1.02084368e-01],\n",
       "         [ 1.00675106e-01,  1.05923682e-01,  1.19140267e-01],\n",
       "         [-1.10364556e-02,  4.48040366e-02,  2.16966271e-02],\n",
       "         ...,\n",
       "         [ 4.76780534e-03,  4.72560525e-03,  7.31605291e-03],\n",
       "         [-9.17375088e-04,  9.76613164e-03, -2.22490728e-02],\n",
       "         [-4.75095809e-02, -2.79402733e-02, -5.42837381e-02]],\n",
       "\n",
       "        [[ 1.13485992e-01,  1.19838238e-01,  1.33462608e-01],\n",
       "         [ 8.39946270e-02,  8.79114270e-02,  1.00319147e-01],\n",
       "         [ 4.74038124e-02,  4.93015051e-02,  5.75497746e-02],\n",
       "         ...,\n",
       "         [ 1.92541182e-02,  3.20705771e-02,  3.58578563e-03],\n",
       "         [ 2.02116370e-03,  1.20221674e-02, -2.07094848e-02],\n",
       "         [ 6.24603033e-03,  1.40821934e-02,  7.01230764e-03]],\n",
       "\n",
       "        [[ 1.41592413e-01,  1.76053375e-01,  1.69002920e-01],\n",
       "         [ 2.45281160e-02,  2.96189487e-02,  3.94488871e-02],\n",
       "         [-6.67691231e-04,  4.19259071e-03,  1.50686502e-02],\n",
       "         ...,\n",
       "         [ 3.01454067e-02,  4.01966870e-02,  8.76775384e-03],\n",
       "         [ 1.03961855e-01,  1.14158124e-01,  8.26086998e-02],\n",
       "         [ 4.99005318e-02,  6.34322762e-02,  2.70447433e-02]]],\n",
       "\n",
       "\n",
       "       [[[ 3.25712115e-01,  2.88693130e-01,  2.87956268e-01],\n",
       "         [ 2.68366307e-01,  2.48930603e-01,  2.60918438e-01],\n",
       "         [ 1.73034668e-02, -1.61476731e-02, -5.05587459e-03],\n",
       "         ...,\n",
       "         [-4.44714904e-01, -4.32071120e-01, -4.26886111e-01],\n",
       "         [-4.33881462e-01, -4.22179639e-01, -4.19832408e-01],\n",
       "         [-4.29961234e-01, -4.20270473e-01, -4.22613829e-01]],\n",
       "\n",
       "        [[-4.41297710e-01, -4.57533836e-01, -4.57365692e-01],\n",
       "         [-4.52583253e-01, -4.67718631e-01, -4.56972510e-01],\n",
       "         [-4.21253920e-01, -4.31243896e-01, -4.34133857e-01],\n",
       "         ...,\n",
       "         [-5.36358237e-01, -5.01490772e-01, -4.82820511e-01],\n",
       "         [-4.13462877e-01, -4.01494354e-01, -4.06648129e-01],\n",
       "         [-4.38642532e-01, -4.34534550e-01, -4.30235386e-01]],\n",
       "\n",
       "        [[-4.48649466e-01, -4.74337131e-01, -4.49353755e-01],\n",
       "         [-4.42008168e-01, -4.52796906e-01, -4.48017925e-01],\n",
       "         [-4.55636859e-01, -4.53163415e-01, -4.54637706e-01],\n",
       "         ...,\n",
       "         [-4.84483480e-01, -4.69045877e-01, -4.58439648e-01],\n",
       "         [-5.24948478e-01, -5.11207819e-01, -4.55043435e-01],\n",
       "         [-3.91313642e-01, -3.95280361e-01, -4.06532586e-01]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[-3.26851308e-01, -2.96087265e-01, -3.41052890e-01],\n",
       "         [-3.22854340e-01, -2.90154755e-01, -3.43604863e-01],\n",
       "         [-3.48291397e-01, -3.19901884e-01, -3.78303409e-01],\n",
       "         ...,\n",
       "         [-4.14840043e-01, -4.22725379e-01, -4.24056500e-01],\n",
       "         [-4.51897770e-01, -4.37292695e-01, -4.41856921e-01],\n",
       "         [-4.63195860e-01, -4.63234395e-01, -4.58205312e-01]],\n",
       "\n",
       "        [[-3.10043454e-01, -2.76240230e-01, -3.29282522e-01],\n",
       "         [-3.08162272e-01, -2.57186621e-01, -3.19288731e-01],\n",
       "         [-2.58478552e-01, -2.29129881e-01, -2.87548304e-01],\n",
       "         ...,\n",
       "         [-4.08196867e-01, -4.14988250e-01, -4.12100494e-01],\n",
       "         [-4.29351389e-01, -4.27193522e-01, -4.28552628e-01],\n",
       "         [-4.72185344e-01, -4.68270779e-01, -4.63575929e-01]],\n",
       "\n",
       "        [[-2.78015435e-01, -2.35711351e-01, -2.89820611e-01],\n",
       "         [-3.28413069e-01, -2.95871258e-01, -3.52707982e-01],\n",
       "         [-2.16353953e-01, -1.72277987e-01, -2.67284274e-01],\n",
       "         ...,\n",
       "         [-4.16913420e-01, -4.18626845e-01, -4.18683231e-01],\n",
       "         [-3.27410728e-01, -3.05449724e-01, -3.36999178e-01],\n",
       "         [-2.20687732e-01, -1.79704994e-01, -2.82759190e-01]]],\n",
       "\n",
       "\n",
       "       [[[-3.37033004e-01, -3.77973557e-01, -3.70867282e-01],\n",
       "         [-3.43398422e-01, -3.70677263e-01, -3.58689427e-01],\n",
       "         [-3.43480855e-01, -3.65167290e-01, -3.61918628e-01],\n",
       "         ...,\n",
       "         [-3.30989420e-01, -3.26188743e-01, -3.13160598e-01],\n",
       "         [-3.39763820e-01, -3.35905135e-01, -3.25714767e-01],\n",
       "         [-3.47608298e-01, -3.45760643e-01, -3.40260893e-01]],\n",
       "\n",
       "        [[-3.51101637e-01, -3.83024037e-01, -3.59326482e-01],\n",
       "         [-3.54544044e-01, -3.77522558e-01, -3.66776437e-01],\n",
       "         [-3.27136278e-01, -3.44969392e-01, -3.47859353e-01],\n",
       "         ...,\n",
       "         [-2.10868031e-01, -2.15216249e-01, -2.59291083e-01],\n",
       "         [-3.42874646e-01, -3.38749230e-01, -3.24295163e-01],\n",
       "         [-3.40603322e-01, -3.40416908e-01, -3.32196176e-01]],\n",
       "\n",
       "        [[-3.42767119e-01, -3.72376353e-01, -3.51314545e-01],\n",
       "         [-3.24361116e-01, -3.46914560e-01, -3.34292442e-01],\n",
       "         [-3.69362354e-01, -3.86496753e-01, -3.76206338e-01],\n",
       "         ...,\n",
       "         [-3.78601134e-01, -4.14143920e-01, -4.19223964e-01],\n",
       "         [ 1.10345662e-01,  6.52627945e-02, -4.32787240e-02],\n",
       "         [-2.69744992e-01, -2.73711741e-01, -2.57512987e-01]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[-3.85674864e-01, -3.94126475e-01, -3.84190172e-01],\n",
       "         [-3.93442571e-01, -4.03880268e-01, -3.94585252e-01],\n",
       "         [-4.22801197e-01, -4.37548935e-01, -4.33205366e-01],\n",
       "         ...,\n",
       "         [-3.36408675e-01, -3.52137148e-01, -3.53468269e-01],\n",
       "         [-3.53858560e-01, -3.66704464e-01, -3.59503984e-01],\n",
       "         [-3.88686061e-01, -4.00489300e-01, -3.87617081e-01]],\n",
       "\n",
       "        [[-3.57102275e-01, -3.66436303e-01, -3.56733501e-01],\n",
       "         [-3.98358345e-01, -4.10127819e-01, -4.01641667e-01],\n",
       "         [-4.19262886e-01, -4.33051467e-01, -4.28724766e-01],\n",
       "         ...,\n",
       "         [-3.18000793e-01, -3.28713745e-01, -3.17982852e-01],\n",
       "         [-3.27390611e-01, -3.40919018e-01, -3.34434986e-01],\n",
       "         [-3.58459830e-01, -3.66310000e-01, -3.53771985e-01]],\n",
       "\n",
       "        [[-3.48603696e-01, -3.57279986e-01, -3.48644167e-01],\n",
       "         [-3.71550351e-01, -3.82145792e-01, -3.76237422e-01],\n",
       "         [-3.81059855e-01, -3.91885847e-01, -3.84931356e-01],\n",
       "         ...,\n",
       "         [-3.03187937e-01, -3.24509203e-01, -3.16722453e-01],\n",
       "         [-3.47018570e-01, -3.60351712e-01, -3.52685452e-01],\n",
       "         [-3.34413230e-01, -3.44410896e-01, -3.41582716e-01]]],\n",
       "\n",
       "\n",
       "       ...,\n",
       "\n",
       "\n",
       "       [[[-2.72290707e-02, -5.42452931e-03,  7.61915743e-02],\n",
       "         [-2.02221930e-01, -1.15775287e-01, -7.24149048e-02],\n",
       "         [-3.36769223e-02, -2.79123783e-02,  6.70883060e-03],\n",
       "         ...,\n",
       "         [ 8.07753205e-02,  6.98896945e-02,  1.02525681e-01],\n",
       "         [ 3.27852070e-02,  3.66439223e-02,  7.82068372e-02],\n",
       "         [ 4.45485711e-02,  5.03177941e-02,  7.93469846e-02]],\n",
       "\n",
       "        [[-8.05133879e-02, -5.75338304e-02,  2.89088488e-02],\n",
       "         [ 2.31871009e-03,  2.24774778e-02,  6.45961463e-02],\n",
       "         [ 1.79617703e-02,  3.15012038e-02,  1.03121072e-01],\n",
       "         ...,\n",
       "         [-3.36358249e-01, -3.09333891e-01, -3.02428365e-01],\n",
       "         [-9.54130292e-03,  6.34878874e-03,  2.86459923e-02],\n",
       "         [ 3.58672738e-02,  3.99752855e-02,  7.56469965e-02]],\n",
       "\n",
       "        [[-3.66296530e-01, -2.78258681e-01, -2.84647882e-01],\n",
       "         [-3.41650248e-02, -9.65964794e-03,  7.35507309e-02],\n",
       "         [-4.65646386e-03, -2.18302011e-03,  6.30093813e-02],\n",
       "         ...,\n",
       "         [-2.13895231e-01, -1.82771355e-01, -2.38831788e-01],\n",
       "         [-3.91615152e-01, -3.62188220e-01, -3.13866973e-01],\n",
       "         [ 9.10393298e-02,  1.10602021e-01,  1.38565481e-01]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[-1.22929752e-01, -1.27459794e-01, -1.09680355e-01],\n",
       "         [-1.07168049e-01, -1.21527314e-01, -1.04389161e-01],\n",
       "         [-1.28683537e-01, -1.39509708e-01, -1.11636728e-01],\n",
       "         ...,\n",
       "         [ 2.12610960e-01,  1.69431508e-01,  1.36727840e-01],\n",
       "         [-8.71918797e-02, -4.90573943e-02,  5.20190597e-03],\n",
       "         [-9.06468332e-02, -1.10293210e-01, -7.78131485e-02]],\n",
       "\n",
       "        [[-3.16120684e-02, -3.70245278e-02, -1.94785893e-02],\n",
       "         [-1.23848528e-01, -1.35618001e-01, -1.07524008e-01],\n",
       "         [-1.25145227e-01, -1.38933808e-01, -1.22842401e-01],\n",
       "         ...,\n",
       "         [-8.27066600e-02, -1.13027453e-01, -1.57198519e-01],\n",
       "         [-8.81749094e-02, -1.17389590e-01, -7.16898739e-02],\n",
       "         [-3.68912220e-02, -8.78786147e-02, -6.35759234e-02]],\n",
       "\n",
       "        [[-3.87997627e-02, -2.78682113e-02, -1.53108239e-02],\n",
       "         [ 9.51163471e-02,  1.11971885e-01,  1.64939076e-01],\n",
       "         [-4.38049436e-02, -6.24740720e-02, -3.98333073e-02],\n",
       "         ...,\n",
       "         [-1.81619301e-01, -1.16666049e-01, -8.53498876e-02],\n",
       "         [-8.03518891e-02,  4.32610512e-04,  1.02216542e-01],\n",
       "         [-1.42256349e-01, -1.91469699e-01, -1.41582698e-01]]],\n",
       "\n",
       "\n",
       "       [[[-3.29189867e-01, -3.62287283e-01, -3.55180979e-01],\n",
       "         [-4.25751358e-01, -4.33422357e-01, -4.17512953e-01],\n",
       "         [-3.98382813e-01, -3.80853564e-01, -3.81526470e-01],\n",
       "         ...,\n",
       "         [ 1.94500804e-01,  2.18909293e-01,  2.31937438e-01],\n",
       "         [-1.55450076e-01, -1.63356096e-01, -1.49244159e-01],\n",
       "         [-1.71137691e-01, -1.81054771e-01, -1.71633422e-01]],\n",
       "\n",
       "        [[-4.25611436e-01, -4.26161289e-01, -4.06385303e-01],\n",
       "         [-3.97681296e-01, -3.93208832e-01, -3.74619573e-01],\n",
       "         [-3.89881372e-01, -3.88106644e-01, -3.83153468e-01],\n",
       "         ...,\n",
       "         [-2.57926852e-01, -2.46588796e-01, -2.00467557e-01],\n",
       "         [-1.35031492e-01, -1.50513947e-01, -1.32138312e-01],\n",
       "         [-1.56289577e-01, -1.67867869e-01, -1.55725569e-01]],\n",
       "\n",
       "        [[-4.01590645e-01, -4.19435173e-01, -3.86608660e-01],\n",
       "         [-3.87106210e-01, -3.90051812e-01, -3.81351262e-01],\n",
       "         [-3.96813333e-01, -3.86496753e-01, -3.68363202e-01],\n",
       "         ...,\n",
       "         [-2.13895231e-01, -2.06300765e-01, -1.72165126e-01],\n",
       "         [-2.62203366e-01, -2.52384275e-01, -1.88376755e-01],\n",
       "         [-1.20725393e-01, -1.40378386e-01, -1.28101200e-01]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[-2.24890530e-01, -2.29420573e-01, -2.07719564e-01],\n",
       "         [-2.05207258e-01, -2.11723387e-01, -1.90663666e-01],\n",
       "         [-7.77031481e-02, -6.10783398e-02, -1.35974884e-02],\n",
       "         ...,\n",
       "         [-7.75851309e-02, -7.76273310e-02, -5.15074730e-02],\n",
       "         [ 3.43767405e-02,  1.15648508e-01,  1.34613693e-01],\n",
       "         [-2.20058590e-01, -2.16175556e-01, -1.99381769e-01]],\n",
       "\n",
       "        [[-1.84553236e-01, -1.89965695e-01, -1.68498188e-01],\n",
       "         [-2.06201464e-01, -2.06206232e-01, -1.85955375e-01],\n",
       "         [-1.01615816e-01, -8.01102817e-02, -2.87247300e-02],\n",
       "         ...,\n",
       "         [ 5.06266654e-02,  4.77568507e-02,  4.28014696e-02],\n",
       "         [-6.07239306e-02, -7.03307688e-02, -5.20820320e-02],\n",
       "         [-2.09440231e-01, -2.09447235e-01, -1.92987680e-01]],\n",
       "\n",
       "        [[-1.72133088e-01, -1.76887810e-01, -1.56487286e-01],\n",
       "         [-1.91158175e-01, -1.89988911e-01, -1.72315836e-01],\n",
       "         [-1.81059837e-01, -1.80121124e-01, -1.61401927e-01],\n",
       "         ...,\n",
       "         [-1.62011445e-01, -1.83332711e-01, -1.52016550e-01],\n",
       "         [-1.39175415e-01, -1.40743852e-01, -1.25234455e-01],\n",
       "         [-1.81472033e-01, -1.83626562e-01, -1.76876813e-01]]],\n",
       "\n",
       "\n",
       "       [[[ 3.96300346e-01,  3.39673519e-01,  3.27171952e-01],\n",
       "         [ 4.21307474e-01,  3.74420792e-01,  3.74643922e-01],\n",
       "         [ 4.32989746e-01,  3.91695470e-01,  3.83179426e-01],\n",
       "         ...,\n",
       "         [-1.50597245e-01, -2.28149533e-01, -2.38650814e-01],\n",
       "         [ 4.40628350e-01,  4.52330172e-01,  4.82128382e-01],\n",
       "         [ 3.70038778e-01,  3.64043266e-01,  4.04837161e-01]],\n",
       "\n",
       "        [[ 4.41055238e-01,  3.93446565e-01,  3.97536278e-01],\n",
       "         [ 4.68985379e-01,  4.26399022e-01,  4.25380439e-01],\n",
       "         [ 4.65020597e-01,  4.27579641e-01,  4.12924975e-01],\n",
       "         ...,\n",
       "         [ 4.32269216e-01,  4.35764134e-01,  4.81885374e-01],\n",
       "         [ 4.13988113e-01,  4.92623299e-01,  5.03155828e-01],\n",
       "         [ 5.02533913e-01,  5.06641924e-01,  5.14862657e-01]],\n",
       "\n",
       "        [[ 4.45468187e-01,  3.96251112e-01,  4.05548215e-01],\n",
       "         [ 3.85442823e-01,  3.47203106e-01,  3.40217382e-01],\n",
       "         [ 3.71814132e-01,  3.58601302e-01,  3.53205442e-01],\n",
       "         ...,\n",
       "         [ 4.72379267e-01,  4.72130597e-01,  5.06266236e-01],\n",
       "         [ 4.27992702e-01,  4.02517676e-01,  2.90054619e-01],\n",
       "         [ 4.43980485e-01,  4.43935335e-01,  4.52290952e-01]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 2.49619275e-01,  2.21559823e-01,  2.19731420e-01],\n",
       "         [ 2.92831957e-01,  2.70629555e-01,  2.44630456e-01],\n",
       "         [ 2.39943922e-01,  2.05588341e-01,  1.98167205e-01],\n",
       "         ...,\n",
       "         [ 2.24375665e-01,  2.12568760e-01,  1.95551366e-01],\n",
       "         [ 1.87317938e-01,  1.90158308e-01,  1.93437219e-01],\n",
       "         [ 1.48568869e-01,  1.64216608e-01,  1.49637848e-01]],\n",
       "\n",
       "        [[ 3.09564412e-01,  2.88465679e-01,  2.62874365e-01],\n",
       "         [ 2.17327952e-01,  1.85950637e-01,  1.82672083e-01],\n",
       "         [ 3.10148895e-01,  2.76752472e-01,  2.69314468e-01],\n",
       "         ...,\n",
       "         [ 2.46705115e-01,  2.51678437e-01,  2.58487761e-01],\n",
       "         [ 3.51040781e-01,  3.53198647e-01,  3.55761111e-01],\n",
       "         [ 4.21932310e-01,  4.25846875e-01,  4.38384861e-01]],\n",
       "\n",
       "        [[ 3.80808085e-01,  3.52523953e-01,  3.49395066e-01],\n",
       "         [ 4.75508481e-01,  4.45305198e-01,  4.39448863e-01],\n",
       "         [ 3.52273494e-01,  3.29682797e-01,  3.01343173e-01],\n",
       "         ...,\n",
       "         [ 3.83086592e-01,  3.65686893e-01,  3.57787371e-01],\n",
       "         [ 3.11804980e-01,  3.33765954e-01,  3.33589077e-01],\n",
       "         [ 3.20488751e-01,  3.34020495e-01,  3.32927108e-01]]]],\n",
       "      dtype=float32)"
      ]
     },
     "execution_count": 29,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_x"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 1/100\n",
      "  1/184 [..............................] - ETA: 0s - loss: 1.7919 - acc: 0.0938WARNING:tensorflow:Callbacks method `on_train_batch_end` is slow compared to the batch time (batch time: 0.0079s vs `on_train_batch_end` time: 0.0150s). Check your callbacks.\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.3562 - acc: 0.5799\n",
      "Epoch 00001: val_acc improved from -inf to 0.59340, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.01-0.59.h5\n",
      "184/184 [==============================] - 7s 35ms/step - loss: 1.3568 - acc: 0.5794 - val_loss: 1.3003 - val_acc: 0.5934\n",
      "Epoch 2/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.2854 - acc: 0.5813\n",
      "Epoch 00002: val_acc did not improve from 0.59340\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 1.2845 - acc: 0.5813 - val_loss: 1.1811 - val_acc: 0.5934\n",
      "Epoch 3/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.1259 - acc: 0.6196\n",
      "Epoch 00003: val_acc improved from 0.59340 to 0.60127, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.03-0.60.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 1.1254 - acc: 0.6196 - val_loss: 1.1199 - val_acc: 0.6013\n",
      "Epoch 4/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.9975 - acc: 0.6529\n",
      "Epoch 00004: val_acc improved from 0.60127 to 0.69975, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.04-0.70.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.9975 - acc: 0.6529 - val_loss: 0.8673 - val_acc: 0.6997\n",
      "Epoch 5/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.8311 - acc: 0.7055\n",
      "Epoch 00005: val_acc improved from 0.69975 to 0.75102, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.05-0.75.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.8311 - acc: 0.7055 - val_loss: 0.7009 - val_acc: 0.7510\n",
      "Epoch 6/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.7007 - acc: 0.7504\n",
      "Epoch 00006: val_acc improved from 0.75102 to 0.77970, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.06-0.78.h5\n",
      "184/184 [==============================] - 6s 35ms/step - loss: 0.7007 - acc: 0.7504 - val_loss: 0.6436 - val_acc: 0.7797\n",
      "Epoch 7/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.6029 - acc: 0.7804\n",
      "Epoch 00007: val_acc improved from 0.77970 to 0.80152, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.07-0.80.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 0.6029 - acc: 0.7804 - val_loss: 0.5886 - val_acc: 0.8015\n",
      "Epoch 8/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.5385 - acc: 0.8108\n",
      "Epoch 00008: val_acc did not improve from 0.80152\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.5385 - acc: 0.8108 - val_loss: 0.5642 - val_acc: 0.8015\n",
      "Epoch 9/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.4736 - acc: 0.8309\n",
      "Epoch 00009: val_acc improved from 0.80152 to 0.81675, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.09-0.82.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 0.4736 - acc: 0.8309 - val_loss: 0.5207 - val_acc: 0.8168\n",
      "Epoch 10/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.4228 - acc: 0.8482\n",
      "Epoch 00010: val_acc improved from 0.81675 to 0.83249, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.10-0.83.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.4228 - acc: 0.8482 - val_loss: 0.4682 - val_acc: 0.8325\n",
      "Epoch 11/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.3803 - acc: 0.8668\n",
      "Epoch 00011: val_acc did not improve from 0.83249\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.3803 - acc: 0.8668 - val_loss: 0.5127 - val_acc: 0.8226\n",
      "Epoch 12/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3229 - acc: 0.8876\n",
      "Epoch 00012: val_acc did not improve from 0.83249\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.3228 - acc: 0.8877 - val_loss: 0.5760 - val_acc: 0.8058\n",
      "Epoch 13/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2863 - acc: 0.9016\n",
      "Epoch 00013: val_acc did not improve from 0.83249\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.2858 - acc: 0.9018 - val_loss: 0.5669 - val_acc: 0.8061\n",
      "Epoch 14/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2570 - acc: 0.9098\n",
      "Epoch 00014: val_acc improved from 0.83249 to 0.84264, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.14-0.84.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 0.2570 - acc: 0.9098 - val_loss: 0.5318 - val_acc: 0.8426\n",
      "Epoch 15/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2132 - acc: 0.9228\n",
      "Epoch 00015: val_acc improved from 0.84264 to 0.85787, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.15-0.86.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.2132 - acc: 0.9228 - val_loss: 0.5065 - val_acc: 0.8579\n",
      "Epoch 16/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.1999 - acc: 0.9311\n",
      "Epoch 00016: val_acc did not improve from 0.85787\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.1999 - acc: 0.9311 - val_loss: 0.5961 - val_acc: 0.8500\n",
      "Epoch 17/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.1748 - acc: 0.9405\n",
      "Epoch 00017: val_acc improved from 0.85787 to 0.85812, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.17-0.86.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.1748 - acc: 0.9405 - val_loss: 0.5575 - val_acc: 0.8581\n",
      "Epoch 18/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.1538 - acc: 0.9477\n",
      "Epoch 00018: val_acc improved from 0.85812 to 0.86066, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.18-0.86.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.1532 - acc: 0.9479 - val_loss: 0.4701 - val_acc: 0.8607\n",
      "Epoch 19/100\n",
      "182/184 [============================>.] - ETA: 0s - loss: 0.1427 - acc: 0.9503\n",
      "Epoch 00019: val_acc did not improve from 0.86066\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.1425 - acc: 0.9507 - val_loss: 0.5464 - val_acc: 0.8272\n",
      "Epoch 20/100\n",
      "182/184 [============================>.] - ETA: 0s - loss: 0.1242 - acc: 0.9565\n",
      "Epoch 00020: val_acc did not improve from 0.86066\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.1256 - acc: 0.9559 - val_loss: 0.5746 - val_acc: 0.8548\n",
      "Epoch 21/100\n",
      "182/184 [============================>.] - ETA: 0s - loss: 0.1244 - acc: 0.9573\n",
      "Epoch 00021: val_acc improved from 0.86066 to 0.86269, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.21-0.86.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.1244 - acc: 0.9573 - val_loss: 0.5689 - val_acc: 0.8627\n",
      "Epoch 22/100\n",
      "182/184 [============================>.] - ETA: 0s - loss: 0.1018 - acc: 0.9647\n",
      "Epoch 00022: val_acc did not improve from 0.86269\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.1037 - acc: 0.9643 - val_loss: 0.6073 - val_acc: 0.8487\n",
      "Epoch 23/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0952 - acc: 0.9644\n",
      "Epoch 00023: val_acc did not improve from 0.86269\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0952 - acc: 0.9644 - val_loss: 0.6478 - val_acc: 0.8345\n",
      "Epoch 24/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0871 - acc: 0.9726\n",
      "Epoch 00024: val_acc improved from 0.86269 to 0.86574, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.24-0.87.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.0871 - acc: 0.9726 - val_loss: 0.5934 - val_acc: 0.8657\n",
      "Epoch 25/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0745 - acc: 0.9743\n",
      "Epoch 00025: val_acc did not improve from 0.86574\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0745 - acc: 0.9743 - val_loss: 0.6778 - val_acc: 0.8520\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 26/100\n",
      "182/184 [============================>.] - ETA: 0s - loss: 0.0870 - acc: 0.9730- ETA: 0s - loss: 0.0885 - acc: 0.\n",
      "Epoch 00026: val_acc did not improve from 0.86574\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0881 - acc: 0.9729 - val_loss: 0.6883 - val_acc: 0.8556\n",
      "Epoch 27/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.0641 - acc: 0.9774\n",
      "Epoch 00027: val_acc improved from 0.86574 to 0.86701, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.27-0.87.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 0.0651 - acc: 0.9770 - val_loss: 0.7251 - val_acc: 0.8670\n",
      "Epoch 28/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0602 - acc: 0.9823\n",
      "Epoch 00028: val_acc improved from 0.86701 to 0.87589, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.28-0.88.h5\n",
      "184/184 [==============================] - 6s 33ms/step - loss: 0.0602 - acc: 0.9823 - val_loss: 0.5819 - val_acc: 0.8759\n",
      "Epoch 29/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.0429 - acc: 0.9877- ETA: 1s - \n",
      "Epoch 00029: val_acc did not improve from 0.87589\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0438 - acc: 0.9874 - val_loss: 0.5840 - val_acc: 0.8614\n",
      "Epoch 30/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.0542 - acc: 0.9819\n",
      "Epoch 00030: val_acc did not improve from 0.87589\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.0544 - acc: 0.9818 - val_loss: 0.6124 - val_acc: 0.8673\n",
      "Epoch 31/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.0532 - acc: 0.9814\n",
      "Epoch 00031: val_acc did not improve from 0.87589\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.0530 - acc: 0.9815 - val_loss: 0.6624 - val_acc: 0.8673\n",
      "Epoch 32/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0471 - acc: 0.9849\n",
      "Epoch 00032: val_acc did not improve from 0.87589\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0471 - acc: 0.9849 - val_loss: 0.7132 - val_acc: 0.8726\n",
      "Epoch 33/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0267 - acc: 0.9927\n",
      "Epoch 00033: val_acc did not improve from 0.87589\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0267 - acc: 0.9927 - val_loss: 0.7054 - val_acc: 0.8675\n",
      "Epoch 34/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0145 - acc: 0.9964\n",
      "Epoch 00034: val_acc did not improve from 0.87589\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0145 - acc: 0.9964 - val_loss: 0.8304 - val_acc: 0.8665\n",
      "Epoch 35/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0442 - acc: 0.9879\n",
      "Epoch 00035: val_acc improved from 0.87589 to 0.87665, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.35-0.88.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 0.0442 - acc: 0.9879 - val_loss: 0.6507 - val_acc: 0.8766\n",
      "Epoch 36/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0362 - acc: 0.9900\n",
      "Epoch 00036: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0362 - acc: 0.9900 - val_loss: 0.7197 - val_acc: 0.8673\n",
      "Epoch 37/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0327 - acc: 0.9896\n",
      "Epoch 00037: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0327 - acc: 0.9896 - val_loss: 0.7136 - val_acc: 0.8596\n",
      "Epoch 38/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0295 - acc: 0.9910\n",
      "Epoch 00038: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0295 - acc: 0.9910 - val_loss: 0.6943 - val_acc: 0.8744\n",
      "Epoch 39/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0228 - acc: 0.9930\n",
      "Epoch 00039: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0228 - acc: 0.9930 - val_loss: 0.6957 - val_acc: 0.8675\n",
      "Epoch 40/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0220 - acc: 0.9932\n",
      "Epoch 00040: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0220 - acc: 0.9932 - val_loss: 0.7768 - val_acc: 0.8662\n",
      "Epoch 41/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0208 - acc: 0.9942\n",
      "Epoch 00041: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0208 - acc: 0.9942 - val_loss: 0.7571 - val_acc: 0.8718\n",
      "Epoch 42/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0316 - acc: 0.9903- ETA: 1s - loss: 0.\n",
      "Epoch 00042: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0316 - acc: 0.9903 - val_loss: 0.7248 - val_acc: 0.8746\n",
      "Epoch 43/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0131 - acc: 0.9968\n",
      "Epoch 00043: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0131 - acc: 0.9968 - val_loss: 0.8532 - val_acc: 0.8632\n",
      "Epoch 44/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0271 - acc: 0.9925\n",
      "Epoch 00044: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0271 - acc: 0.9925 - val_loss: 0.7922 - val_acc: 0.8728\n",
      "Epoch 45/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0163 - acc: 0.9949\n",
      "Epoch 00045: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.0163 - acc: 0.9949 - val_loss: 0.8547 - val_acc: 0.8759\n",
      "Epoch 46/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.0209 - acc: 0.9928\n",
      "Epoch 00046: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0208 - acc: 0.9929 - val_loss: 0.9718 - val_acc: 0.8439\n",
      "Epoch 47/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0429 - acc: 0.9883- ETA: 1s - loss: 0\n",
      "Epoch 00047: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0429 - acc: 0.9883 - val_loss: 0.6286 - val_acc: 0.8716\n",
      "Epoch 48/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0072 - acc: 0.9976\n",
      "Epoch 00048: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0072 - acc: 0.9976 - val_loss: 0.7970 - val_acc: 0.8731\n",
      "Epoch 49/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0060 - acc: 0.9985- ETA: 1s - loss: 0.00\n",
      "Epoch 00049: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0060 - acc: 0.9985 - val_loss: 0.8904 - val_acc: 0.8739\n",
      "Epoch 50/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.0173 - acc: 0.9940\n",
      "Epoch 00050: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 0.0173 - acc: 0.9940 - val_loss: 0.6522 - val_acc: 0.8734\n",
      "Epoch 51/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0088 - acc: 0.9978\n",
      "Epoch 00051: val_acc did not improve from 0.87665\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 0.0088 - acc: 0.9978 - val_loss: 0.7562 - val_acc: 0.8766\n",
      "Epoch 52/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.0017 - acc: 0.9993\n",
      "Epoch 00052: val_acc improved from 0.87665 to 0.87792, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.52-0.88.h5\n",
      "184/184 [==============================] - 6s 32ms/step - loss: 0.0017 - acc: 0.9993 - val_loss: 0.8585 - val_acc: 0.8779\n",
      "Epoch 53/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 6.5715e-04 - acc: 0.9995\n",
      "Epoch 00053: val_acc improved from 0.87792 to 0.88198, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\vgg19_model.53-0.88.h5\n",
      "184/184 [==============================] - 6s 34ms/step - loss: 6.5715e-04 - acc: 0.9995 - val_loss: 0.8742 - val_acc: 0.8820\n",
      "Epoch 54/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 4.2960e-04 - acc: 0.9998\n",
      "Epoch 00054: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 4.2960e-04 - acc: 0.9998 - val_loss: 0.8960 - val_acc: 0.8817\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 55/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.2229e-04 - acc: 0.9998\n",
      "Epoch 00055: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 3.2229e-04 - acc: 0.9998 - val_loss: 0.9025 - val_acc: 0.8807\n",
      "Epoch 56/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.0566e-04 - acc: 1.0000\n",
      "Epoch 00056: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 28ms/step - loss: 1.0512e-04 - acc: 1.0000 - val_loss: 0.9227 - val_acc: 0.8805\n",
      "Epoch 57/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 4.3181e-04 - acc: 0.9997\n",
      "Epoch 00057: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 4.2966e-04 - acc: 0.9997 - val_loss: 0.9367 - val_acc: 0.8807\n",
      "Epoch 58/100\n",
      "183/184 [============================>.] - ETA: 0s - loss: 2.9978e-04 - acc: 0.9998- ETA: \n",
      "Epoch 00058: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 2.9868e-04 - acc: 0.9998 - val_loss: 0.9515 - val_acc: 0.8807\n",
      "Epoch 59/100\n",
      "182/184 [============================>.] - ETA: 0s - loss: 4.5107e-04 - acc: 0.9997- ETA: 0s - loss: 4.6516e-04 - acc: 0.99\n",
      "Epoch 00059: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 4.4641e-04 - acc: 0.9997 - val_loss: 0.9475 - val_acc: 0.8810\n",
      "Epoch 60/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 4.2800e-04 - acc: 0.9997- ETA: 2s - \n",
      "Epoch 00060: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 4.2800e-04 - acc: 0.9997 - val_loss: 0.9509 - val_acc: 0.8807\n",
      "Epoch 61/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 4.2235e-04 - acc: 0.9998\n",
      "Epoch 00061: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 4.2235e-04 - acc: 0.9998 - val_loss: 0.9586 - val_acc: 0.8810\n",
      "Epoch 62/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.5179e-04 - acc: 0.9997- ETA: 0s - loss: 3.6258e-04 - acc: 0.99\n",
      "Epoch 00062: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 3.5179e-04 - acc: 0.9997 - val_loss: 0.9684 - val_acc: 0.8812\n",
      "Epoch 63/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 2.9150e-04 - acc: 0.9998\n",
      "Epoch 00063: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 2.9150e-04 - acc: 0.9998 - val_loss: 0.9825 - val_acc: 0.8812\n",
      "Epoch 64/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.4918e-04 - acc: 0.9997\n",
      "Epoch 00064: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 3.4918e-04 - acc: 0.9997 - val_loss: 0.9863 - val_acc: 0.8815\n",
      "Epoch 65/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 2.9764e-04 - acc: 0.9998- ETA: 0s - loss: 3.4627e-04 - ac\n",
      "Epoch 00065: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 2.9764e-04 - acc: 0.9998 - val_loss: 0.9928 - val_acc: 0.8812\n",
      "Epoch 66/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.0938e-04 - acc: 0.9998\n",
      "Epoch 00066: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 3.0938e-04 - acc: 0.9998 - val_loss: 0.9997 - val_acc: 0.8810\n",
      "Epoch 67/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 2.5412e-04 - acc: 0.9998\n",
      "Epoch 00067: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 2.5412e-04 - acc: 0.9998 - val_loss: 1.0101 - val_acc: 0.8807\n",
      "Epoch 68/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 1.8264e-04 - acc: 1.0000\n",
      "Epoch 00068: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 1.8264e-04 - acc: 1.0000 - val_loss: 1.0253 - val_acc: 0.8805\n",
      "Epoch 69/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.6140e-04 - acc: 0.9997\n",
      "Epoch 00069: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 3.6140e-04 - acc: 0.9997 - val_loss: 1.0211 - val_acc: 0.8812\n",
      "Epoch 70/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.5300e-04 - acc: 0.9997\n",
      "Epoch 00070: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 3.5300e-04 - acc: 0.9997 - val_loss: 1.0202 - val_acc: 0.8807\n",
      "Epoch 71/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 4.0123e-04 - acc: 0.9997\n",
      "Epoch 00071: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 4.0123e-04 - acc: 0.9997 - val_loss: 1.0139 - val_acc: 0.8802\n",
      "Epoch 72/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 3.3698e-04 - acc: 0.9997\n",
      "Epoch 00072: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 3.3698e-04 - acc: 0.9997 - val_loss: 1.0218 - val_acc: 0.8805\n",
      "Epoch 73/100\n",
      "184/184 [==============================] - ETA: 0s - loss: 2.6646e-04 - acc: 0.9997Restoring model weights from the end of the best epoch.\n",
      "\n",
      "Epoch 00073: val_acc did not improve from 0.88198\n",
      "184/184 [==============================] - 5s 27ms/step - loss: 2.6646e-04 - acc: 0.9997 - val_loss: 1.0305 - val_acc: 0.8807\n",
      "Epoch 00073: early stopping\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "<tensorflow.python.keras.callbacks.History at 0x267bd249d30>"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "batch_size = 32\n",
    "epochs = 100\n",
    "iterations = 184\n",
    "model.fit_generator(datagen.flow(train_x, train_y, batch_size=batch_size),\n",
    "                    steps_per_epoch=iterations,\n",
    "                    epochs=epochs,\n",
    "                    callbacks=cbks,\n",
    "                    validation_data=valid_data_gen.flow(test_x, test_y, batch_size=batch_size))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "def scheduler(epoch):\n",
    "    if epoch  == 50 and epoch != 0:\n",
    "        lr = K.get_value(ResNet_model.optimizer.lr)\n",
    "        K.set_value(ResNet_model.optimizer.lr, lr * 0.1)\n",
    "        print(\"lr changed to {}\".format(lr * 0.1))\n",
    "    if epoch  == 100 and epoch != 0:\n",
    "        lr = K.get_value(ResNet_model.optimizer.lr)\n",
    "        K.set_value(ResNet_model.optimizer.lr, lr * 0.1)\n",
    "        print(\"lr changed to {}\".format(lr * 0.1))\n",
    "    if epoch  == 150 and epoch != 0:\n",
    "        lr = K.get_value(ResNet_model.optimizer.lr)\n",
    "        K.set_value(ResNet_model.optimizer.lr, lr * 0.1)\n",
    "        print(\"lr changed to {}\".format(lr * 0.1))\n",
    "    return  K.get_value(ResNet_model.optimizer.lr)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "input_shape=(32,32,3)\n",
    "num_class = 6\n",
    "ResNet_base = ResNet50(include_top=False, weights='imagenet',\n",
    "                              input_shape=(input_shape[0],input_shape[1],3), pooling='max')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Model: \"functional_1\"\n",
      "__________________________________________________________________________________________________\n",
      "Layer (type)                    Output Shape         Param #     Connected to                     \n",
      "==================================================================================================\n",
      "input_2 (InputLayer)            [(None, 32, 32, 3)]  0                                            \n",
      "__________________________________________________________________________________________________\n",
      "conv1_pad (ZeroPadding2D)       (None, 38, 38, 3)    0           input_2[0][0]                    \n",
      "__________________________________________________________________________________________________\n",
      "conv1_conv (Conv2D)             (None, 16, 16, 64)   9472        conv1_pad[0][0]                  \n",
      "__________________________________________________________________________________________________\n",
      "conv1_bn (BatchNormalization)   (None, 16, 16, 64)   256         conv1_conv[0][0]                 \n",
      "__________________________________________________________________________________________________\n",
      "conv1_relu (Activation)         (None, 16, 16, 64)   0           conv1_bn[0][0]                   \n",
      "__________________________________________________________________________________________________\n",
      "pool1_pad (ZeroPadding2D)       (None, 18, 18, 64)   0           conv1_relu[0][0]                 \n",
      "__________________________________________________________________________________________________\n",
      "pool1_pool (MaxPooling2D)       (None, 8, 8, 64)     0           pool1_pad[0][0]                  \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_1_conv (Conv2D)    (None, 8, 8, 64)     4160        pool1_pool[0][0]                 \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_1_bn (BatchNormali (None, 8, 8, 64)     256         conv2_block1_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_1_relu (Activation (None, 8, 8, 64)     0           conv2_block1_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_2_conv (Conv2D)    (None, 8, 8, 64)     36928       conv2_block1_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_2_bn (BatchNormali (None, 8, 8, 64)     256         conv2_block1_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_2_relu (Activation (None, 8, 8, 64)     0           conv2_block1_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_0_conv (Conv2D)    (None, 8, 8, 256)    16640       pool1_pool[0][0]                 \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_3_conv (Conv2D)    (None, 8, 8, 256)    16640       conv2_block1_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_0_bn (BatchNormali (None, 8, 8, 256)    1024        conv2_block1_0_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_3_bn (BatchNormali (None, 8, 8, 256)    1024        conv2_block1_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_add (Add)          (None, 8, 8, 256)    0           conv2_block1_0_bn[0][0]          \n",
      "                                                                 conv2_block1_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block1_out (Activation)   (None, 8, 8, 256)    0           conv2_block1_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_1_conv (Conv2D)    (None, 8, 8, 64)     16448       conv2_block1_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_1_bn (BatchNormali (None, 8, 8, 64)     256         conv2_block2_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_1_relu (Activation (None, 8, 8, 64)     0           conv2_block2_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_2_conv (Conv2D)    (None, 8, 8, 64)     36928       conv2_block2_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_2_bn (BatchNormali (None, 8, 8, 64)     256         conv2_block2_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_2_relu (Activation (None, 8, 8, 64)     0           conv2_block2_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_3_conv (Conv2D)    (None, 8, 8, 256)    16640       conv2_block2_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_3_bn (BatchNormali (None, 8, 8, 256)    1024        conv2_block2_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_add (Add)          (None, 8, 8, 256)    0           conv2_block1_out[0][0]           \n",
      "                                                                 conv2_block2_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block2_out (Activation)   (None, 8, 8, 256)    0           conv2_block2_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_1_conv (Conv2D)    (None, 8, 8, 64)     16448       conv2_block2_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_1_bn (BatchNormali (None, 8, 8, 64)     256         conv2_block3_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_1_relu (Activation (None, 8, 8, 64)     0           conv2_block3_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_2_conv (Conv2D)    (None, 8, 8, 64)     36928       conv2_block3_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_2_bn (BatchNormali (None, 8, 8, 64)     256         conv2_block3_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_2_relu (Activation (None, 8, 8, 64)     0           conv2_block3_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_3_conv (Conv2D)    (None, 8, 8, 256)    16640       conv2_block3_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_3_bn (BatchNormali (None, 8, 8, 256)    1024        conv2_block3_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_add (Add)          (None, 8, 8, 256)    0           conv2_block2_out[0][0]           \n",
      "                                                                 conv2_block3_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv2_block3_out (Activation)   (None, 8, 8, 256)    0           conv2_block3_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_1_conv (Conv2D)    (None, 4, 4, 128)    32896       conv2_block3_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_1_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block1_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_1_relu (Activation (None, 4, 4, 128)    0           conv3_block1_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_2_conv (Conv2D)    (None, 4, 4, 128)    147584      conv3_block1_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_2_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block1_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_2_relu (Activation (None, 4, 4, 128)    0           conv3_block1_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_0_conv (Conv2D)    (None, 4, 4, 512)    131584      conv2_block3_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_3_conv (Conv2D)    (None, 4, 4, 512)    66048       conv3_block1_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_0_bn (BatchNormali (None, 4, 4, 512)    2048        conv3_block1_0_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_3_bn (BatchNormali (None, 4, 4, 512)    2048        conv3_block1_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_add (Add)          (None, 4, 4, 512)    0           conv3_block1_0_bn[0][0]          \n",
      "                                                                 conv3_block1_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block1_out (Activation)   (None, 4, 4, 512)    0           conv3_block1_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_1_conv (Conv2D)    (None, 4, 4, 128)    65664       conv3_block1_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_1_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block2_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_1_relu (Activation (None, 4, 4, 128)    0           conv3_block2_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_2_conv (Conv2D)    (None, 4, 4, 128)    147584      conv3_block2_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_2_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block2_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_2_relu (Activation (None, 4, 4, 128)    0           conv3_block2_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_3_conv (Conv2D)    (None, 4, 4, 512)    66048       conv3_block2_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_3_bn (BatchNormali (None, 4, 4, 512)    2048        conv3_block2_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_add (Add)          (None, 4, 4, 512)    0           conv3_block1_out[0][0]           \n",
      "                                                                 conv3_block2_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block2_out (Activation)   (None, 4, 4, 512)    0           conv3_block2_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_1_conv (Conv2D)    (None, 4, 4, 128)    65664       conv3_block2_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_1_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block3_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_1_relu (Activation (None, 4, 4, 128)    0           conv3_block3_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_2_conv (Conv2D)    (None, 4, 4, 128)    147584      conv3_block3_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_2_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block3_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_2_relu (Activation (None, 4, 4, 128)    0           conv3_block3_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_3_conv (Conv2D)    (None, 4, 4, 512)    66048       conv3_block3_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_3_bn (BatchNormali (None, 4, 4, 512)    2048        conv3_block3_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_add (Add)          (None, 4, 4, 512)    0           conv3_block2_out[0][0]           \n",
      "                                                                 conv3_block3_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block3_out (Activation)   (None, 4, 4, 512)    0           conv3_block3_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_1_conv (Conv2D)    (None, 4, 4, 128)    65664       conv3_block3_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_1_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block4_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_1_relu (Activation (None, 4, 4, 128)    0           conv3_block4_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_2_conv (Conv2D)    (None, 4, 4, 128)    147584      conv3_block4_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_2_bn (BatchNormali (None, 4, 4, 128)    512         conv3_block4_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_2_relu (Activation (None, 4, 4, 128)    0           conv3_block4_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_3_conv (Conv2D)    (None, 4, 4, 512)    66048       conv3_block4_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_3_bn (BatchNormali (None, 4, 4, 512)    2048        conv3_block4_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_add (Add)          (None, 4, 4, 512)    0           conv3_block3_out[0][0]           \n",
      "                                                                 conv3_block4_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv3_block4_out (Activation)   (None, 4, 4, 512)    0           conv3_block4_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_1_conv (Conv2D)    (None, 2, 2, 256)    131328      conv3_block4_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_1_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block1_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_1_relu (Activation (None, 2, 2, 256)    0           conv4_block1_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_2_conv (Conv2D)    (None, 2, 2, 256)    590080      conv4_block1_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_2_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block1_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_2_relu (Activation (None, 2, 2, 256)    0           conv4_block1_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_0_conv (Conv2D)    (None, 2, 2, 1024)   525312      conv3_block4_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_3_conv (Conv2D)    (None, 2, 2, 1024)   263168      conv4_block1_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_0_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block1_0_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_3_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block1_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_add (Add)          (None, 2, 2, 1024)   0           conv4_block1_0_bn[0][0]          \n",
      "                                                                 conv4_block1_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block1_out (Activation)   (None, 2, 2, 1024)   0           conv4_block1_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_1_conv (Conv2D)    (None, 2, 2, 256)    262400      conv4_block1_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_1_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block2_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_1_relu (Activation (None, 2, 2, 256)    0           conv4_block2_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_2_conv (Conv2D)    (None, 2, 2, 256)    590080      conv4_block2_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_2_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block2_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_2_relu (Activation (None, 2, 2, 256)    0           conv4_block2_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_3_conv (Conv2D)    (None, 2, 2, 1024)   263168      conv4_block2_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_3_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block2_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_add (Add)          (None, 2, 2, 1024)   0           conv4_block1_out[0][0]           \n",
      "                                                                 conv4_block2_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block2_out (Activation)   (None, 2, 2, 1024)   0           conv4_block2_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_1_conv (Conv2D)    (None, 2, 2, 256)    262400      conv4_block2_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_1_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block3_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_1_relu (Activation (None, 2, 2, 256)    0           conv4_block3_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_2_conv (Conv2D)    (None, 2, 2, 256)    590080      conv4_block3_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_2_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block3_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_2_relu (Activation (None, 2, 2, 256)    0           conv4_block3_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_3_conv (Conv2D)    (None, 2, 2, 1024)   263168      conv4_block3_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_3_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block3_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_add (Add)          (None, 2, 2, 1024)   0           conv4_block2_out[0][0]           \n",
      "                                                                 conv4_block3_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block3_out (Activation)   (None, 2, 2, 1024)   0           conv4_block3_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_1_conv (Conv2D)    (None, 2, 2, 256)    262400      conv4_block3_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_1_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block4_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_1_relu (Activation (None, 2, 2, 256)    0           conv4_block4_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_2_conv (Conv2D)    (None, 2, 2, 256)    590080      conv4_block4_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_2_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block4_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_2_relu (Activation (None, 2, 2, 256)    0           conv4_block4_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_3_conv (Conv2D)    (None, 2, 2, 1024)   263168      conv4_block4_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_3_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block4_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_add (Add)          (None, 2, 2, 1024)   0           conv4_block3_out[0][0]           \n",
      "                                                                 conv4_block4_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block4_out (Activation)   (None, 2, 2, 1024)   0           conv4_block4_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_1_conv (Conv2D)    (None, 2, 2, 256)    262400      conv4_block4_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_1_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block5_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_1_relu (Activation (None, 2, 2, 256)    0           conv4_block5_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_2_conv (Conv2D)    (None, 2, 2, 256)    590080      conv4_block5_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_2_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block5_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_2_relu (Activation (None, 2, 2, 256)    0           conv4_block5_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_3_conv (Conv2D)    (None, 2, 2, 1024)   263168      conv4_block5_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_3_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block5_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_add (Add)          (None, 2, 2, 1024)   0           conv4_block4_out[0][0]           \n",
      "                                                                 conv4_block5_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block5_out (Activation)   (None, 2, 2, 1024)   0           conv4_block5_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_1_conv (Conv2D)    (None, 2, 2, 256)    262400      conv4_block5_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_1_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block6_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_1_relu (Activation (None, 2, 2, 256)    0           conv4_block6_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_2_conv (Conv2D)    (None, 2, 2, 256)    590080      conv4_block6_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_2_bn (BatchNormali (None, 2, 2, 256)    1024        conv4_block6_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_2_relu (Activation (None, 2, 2, 256)    0           conv4_block6_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_3_conv (Conv2D)    (None, 2, 2, 1024)   263168      conv4_block6_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_3_bn (BatchNormali (None, 2, 2, 1024)   4096        conv4_block6_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_add (Add)          (None, 2, 2, 1024)   0           conv4_block5_out[0][0]           \n",
      "                                                                 conv4_block6_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv4_block6_out (Activation)   (None, 2, 2, 1024)   0           conv4_block6_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_1_conv (Conv2D)    (None, 1, 1, 512)    524800      conv4_block6_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_1_bn (BatchNormali (None, 1, 1, 512)    2048        conv5_block1_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_1_relu (Activation (None, 1, 1, 512)    0           conv5_block1_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_2_conv (Conv2D)    (None, 1, 1, 512)    2359808     conv5_block1_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_2_bn (BatchNormali (None, 1, 1, 512)    2048        conv5_block1_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_2_relu (Activation (None, 1, 1, 512)    0           conv5_block1_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_0_conv (Conv2D)    (None, 1, 1, 2048)   2099200     conv4_block6_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_3_conv (Conv2D)    (None, 1, 1, 2048)   1050624     conv5_block1_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_0_bn (BatchNormali (None, 1, 1, 2048)   8192        conv5_block1_0_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_3_bn (BatchNormali (None, 1, 1, 2048)   8192        conv5_block1_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_add (Add)          (None, 1, 1, 2048)   0           conv5_block1_0_bn[0][0]          \n",
      "                                                                 conv5_block1_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block1_out (Activation)   (None, 1, 1, 2048)   0           conv5_block1_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_1_conv (Conv2D)    (None, 1, 1, 512)    1049088     conv5_block1_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_1_bn (BatchNormali (None, 1, 1, 512)    2048        conv5_block2_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_1_relu (Activation (None, 1, 1, 512)    0           conv5_block2_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_2_conv (Conv2D)    (None, 1, 1, 512)    2359808     conv5_block2_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_2_bn (BatchNormali (None, 1, 1, 512)    2048        conv5_block2_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_2_relu (Activation (None, 1, 1, 512)    0           conv5_block2_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_3_conv (Conv2D)    (None, 1, 1, 2048)   1050624     conv5_block2_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_3_bn (BatchNormali (None, 1, 1, 2048)   8192        conv5_block2_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_add (Add)          (None, 1, 1, 2048)   0           conv5_block1_out[0][0]           \n",
      "                                                                 conv5_block2_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block2_out (Activation)   (None, 1, 1, 2048)   0           conv5_block2_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_1_conv (Conv2D)    (None, 1, 1, 512)    1049088     conv5_block2_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_1_bn (BatchNormali (None, 1, 1, 512)    2048        conv5_block3_1_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_1_relu (Activation (None, 1, 1, 512)    0           conv5_block3_1_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_2_conv (Conv2D)    (None, 1, 1, 512)    2359808     conv5_block3_1_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_2_bn (BatchNormali (None, 1, 1, 512)    2048        conv5_block3_2_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_2_relu (Activation (None, 1, 1, 512)    0           conv5_block3_2_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_3_conv (Conv2D)    (None, 1, 1, 2048)   1050624     conv5_block3_2_relu[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_3_bn (BatchNormali (None, 1, 1, 2048)   8192        conv5_block3_3_conv[0][0]        \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_add (Add)          (None, 1, 1, 2048)   0           conv5_block2_out[0][0]           \n",
      "                                                                 conv5_block3_3_bn[0][0]          \n",
      "__________________________________________________________________________________________________\n",
      "conv5_block3_out (Activation)   (None, 1, 1, 2048)   0           conv5_block3_add[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "max_pool (GlobalMaxPooling2D)   (None, 2048)         0           conv5_block3_out[0][0]           \n",
      "__________________________________________________________________________________________________\n",
      "flatten (Flatten)               (None, 2048)         0           max_pool[0][0]                   \n",
      "__________________________________________________________________________________________________\n",
      "dense_1 (Dense)                 (None, 512)          1049088     flatten[0][0]                    \n",
      "__________________________________________________________________________________________________\n",
      "dense_2 (Dense)                 (None, 6)            3078        dense_1[0][0]                    \n",
      "==================================================================================================\n",
      "Total params: 24,639,878\n",
      "Trainable params: 24,586,758\n",
      "Non-trainable params: 53,120\n",
      "__________________________________________________________________________________________________\n"
     ]
    }
   ],
   "source": [
    "x = ResNet_base.output\n",
    "x = Flatten(name='flatten')(x)\n",
    "x = Dense(512, activation='relu')(x)\n",
    "# x = BatchNormalization(name='bn_fc_01')(x)\n",
    "# x = Dropout(0.5)(x)\n",
    "predictions = Dense(num_class, activation='softmax')(x)\n",
    "ResNet_model = Model(inputs=ResNet_base.input, outputs=predictions)\n",
    "\n",
    "ResNet_model.summary()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    " GF_model = clone_model(ResNet_model)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'conv5_block1_3_conv'"
      ]
     },
     "execution_count": 37,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "GF_model.layers[150].name"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(2,)"
      ]
     },
     "execution_count": 29,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "np.array(GF_model.layers[178].get_weights()).shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "2"
      ]
     },
     "execution_count": 34,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(GF_model.layers[178].get_weights())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "conv1_conv\n",
      "conv2_block1_1_conv\n",
      "conv2_block1_2_conv\n",
      "conv2_block1_0_conv\n",
      "conv2_block1_3_conv\n",
      "conv2_block2_1_conv\n",
      "conv2_block2_2_conv\n",
      "conv2_block2_3_conv\n",
      "conv2_block3_1_conv\n",
      "conv2_block3_2_conv\n",
      "conv2_block3_3_conv\n",
      "conv3_block1_1_conv\n",
      "conv3_block1_2_conv\n",
      "conv3_block1_0_conv\n",
      "conv3_block1_3_conv\n",
      "conv3_block2_1_conv\n",
      "conv3_block2_2_conv\n",
      "conv3_block2_3_conv\n",
      "conv3_block3_1_conv\n",
      "conv3_block3_2_conv\n",
      "conv3_block3_3_conv\n",
      "conv3_block4_1_conv\n",
      "conv3_block4_2_conv\n",
      "conv3_block4_3_conv\n",
      "conv4_block1_1_conv\n",
      "conv4_block1_2_conv\n",
      "conv4_block1_0_conv\n",
      "conv4_block1_3_conv\n",
      "conv4_block2_1_conv\n",
      "conv4_block2_2_conv\n",
      "conv4_block2_3_conv\n",
      "conv4_block3_1_conv\n",
      "conv4_block3_2_conv\n",
      "conv4_block3_3_conv\n",
      "conv4_block4_1_conv\n",
      "conv4_block4_2_conv\n",
      "conv4_block4_3_conv\n",
      "conv4_block5_1_conv\n",
      "conv4_block5_2_conv\n",
      "conv4_block5_3_conv\n",
      "conv4_block6_1_conv\n",
      "conv4_block6_2_conv\n",
      "conv4_block6_3_conv\n",
      "conv5_block1_1_conv\n",
      "conv5_block1_2_conv\n",
      "conv5_block1_0_conv\n",
      "conv5_block1_3_conv\n",
      "conv5_block2_1_conv\n",
      "conv5_block2_2_conv\n",
      "conv5_block2_3_conv\n",
      "conv5_block3_1_conv\n",
      "conv5_block3_2_conv\n",
      "conv5_block3_3_conv\n",
      "dense_1\n",
      "dense_2\n"
     ]
    }
   ],
   "source": [
    "for layer in GF_model.layers[1:]:\n",
    "    try:\n",
    "        weights = layer.get_weights()\n",
    "#         print(len(weights))\n",
    "        if (len(weights) == 2):\n",
    "            print(layer.name)\n",
    "    except:\n",
    "        pass"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "ename": "AttributeError",
     "evalue": "Can't set the attribute \"name\", likely because it conflicts with an existing read-only @property of the object. Please choose a different name.",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[1;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
      "\u001b[1;32mD:\\Anaconda3\\envs\\grad\\lib\\site-packages\\tensorflow\\python\\keras\\engine\\base_layer.py\u001b[0m in \u001b[0;36m__setattr__\u001b[1;34m(self, name, value)\u001b[0m\n\u001b[0;32m   2761\u001b[0m       \u001b[1;32mtry\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m-> 2762\u001b[1;33m         \u001b[0msuper\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mtracking\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mAutoTrackable\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mself\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0m__setattr__\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mname\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mvalue\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[0;32m   2763\u001b[0m       \u001b[1;32mexcept\u001b[0m \u001b[0mAttributeError\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
      "\u001b[1;31mAttributeError\u001b[0m: can't set attribute",
      "\nDuring handling of the above exception, another exception occurred:\n",
      "\u001b[1;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
      "\u001b[1;32m<ipython-input-20-8978030e45d4>\u001b[0m in \u001b[0;36m<module>\u001b[1;34m\u001b[0m\n\u001b[1;32m----> 1\u001b[1;33m \u001b[0mResNet_model\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mlayers\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;36m0\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mname\u001b[0m \u001b[1;33m+=\u001b[0m \u001b[1;34m'_COPY'\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m",
      "\u001b[1;32mD:\\Anaconda3\\envs\\grad\\lib\\site-packages\\tensorflow\\python\\keras\\engine\\base_layer.py\u001b[0m in \u001b[0;36m__setattr__\u001b[1;34m(self, name, value)\u001b[0m\n\u001b[0;32m   2765\u001b[0m             ('Can\\'t set the attribute \"{}\", likely because it conflicts with '\n\u001b[0;32m   2766\u001b[0m              \u001b[1;34m'an existing read-only @property of the object. Please choose a '\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m-> 2767\u001b[1;33m              'different name.').format(name))\n\u001b[0m\u001b[0;32m   2768\u001b[0m       \u001b[1;32mreturn\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m   2769\u001b[0m \u001b[1;33m\u001b[0m\u001b[0m\n",
      "\u001b[1;31mAttributeError\u001b[0m: Can't set the attribute \"name\", likely because it conflicts with an existing read-only @property of the object. Please choose a different name."
     ]
    }
   ],
   "source": [
    "ResNet_model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "sgd = optimizers.SGD(lr=0.01, decay=1e-4, momentum=0.9, nesterov=True)\n",
    "ResNet_model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['acc'])\n",
    "\n",
    "change_lr = LearningRateScheduler(scheduler)\n",
    "\n",
    "save_dir = os.path.join(os.getcwd(), 'data/trained_model')\n",
    "model_name = 'resnet50_model.{epoch:02d}-{val_acc:.2f}.h5'\n",
    "if not os.path.isdir(save_dir):\n",
    "    os.makedirs(save_dir)\n",
    "filepath = os.path.join(save_dir, model_name)\n",
    "\n",
    "checkpoint = ModelCheckpoint(filepath=filepath,\n",
    "                             monitor='val_acc',\n",
    "                             verbose=1,\n",
    "                             save_best_only=True)\n",
    "\n",
    "#creating early stopping to prevent model from overfitting \n",
    "early_stopping = EarlyStopping(monitor=\"val_acc\", min_delta=0,\n",
    "                                                  patience=30, verbose=1, \n",
    "                                                  mode=\"auto\", baseline=None, \n",
    "                                                  restore_best_weights=True)\n",
    "\n",
    "cbks = [early_stopping, checkpoint, change_lr]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_data_gen = ImageDataGenerator(featurewise_center=True,\n",
    "                                   rotation_range=20,\n",
    "                                   width_shift_range=0.2,\n",
    "                                   height_shift_range=0.2,\n",
    "                                   horizontal_flip=True)\n",
    "valid_data_gen = ImageDataGenerator()\n",
    "train_data_gen.fit(train_x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 1/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.3258 - acc: 0.5927\n",
      "Epoch 00001: val_acc improved from -inf to 0.58985, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.01-0.59.h5\n",
      "184/184 [==============================] - 10s 52ms/step - loss: 1.3244 - acc: 0.5932 - val_loss: 3.5329 - val_acc: 0.5898\n",
      "Epoch 2/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.0681 - acc: 0.6372\n",
      "Epoch 00002: val_acc did not improve from 0.58985\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 1.0676 - acc: 0.6375 - val_loss: 1.7839 - val_acc: 0.5898\n",
      "Epoch 3/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 1.0007 - acc: 0.6558-\n",
      "Epoch 00003: val_acc did not improve from 0.58985\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 1.0002 - acc: 0.6563 - val_loss: 1.6792 - val_acc: 0.5898\n",
      "Epoch 4/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.9818 - acc: 0.6586\n",
      "Epoch 00004: val_acc did not improve from 0.58985\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.9814 - acc: 0.6587 - val_loss: 2.4926 - val_acc: 0.5898\n",
      "Epoch 5/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.9611 - acc: 0.6591\n",
      "Epoch 00005: val_acc improved from 0.58985 to 0.60178, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.05-0.60.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.9606 - acc: 0.6592 - val_loss: 1.7767 - val_acc: 0.6018\n",
      "Epoch 6/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.9073 - acc: 0.6788\n",
      "Epoch 00006: val_acc improved from 0.60178 to 0.66624, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.06-0.67.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.9073 - acc: 0.6788 - val_loss: 1.0581 - val_acc: 0.6662\n",
      "Epoch 7/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.8689 - acc: 0.6967\n",
      "Epoch 00007: val_acc improved from 0.66624 to 0.69264, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.07-0.69.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.8698 - acc: 0.6962 - val_loss: 0.8396 - val_acc: 0.6926\n",
      "Epoch 8/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.8787 - acc: 0.6900\n",
      "Epoch 00008: val_acc did not improve from 0.69264\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.8772 - acc: 0.6907 - val_loss: 0.9174 - val_acc: 0.6876\n",
      "Epoch 9/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.8563 - acc: 0.6943\n",
      "Epoch 00009: val_acc improved from 0.69264 to 0.74518, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.09-0.75.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.8563 - acc: 0.6943 - val_loss: 0.7213 - val_acc: 0.7452\n",
      "Epoch 10/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.8075 - acc: 0.7127\n",
      "Epoch 00010: val_acc did not improve from 0.74518\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.8075 - acc: 0.7127 - val_loss: 0.8145 - val_acc: 0.7249\n",
      "Epoch 11/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.7730 - acc: 0.7324\n",
      "Epoch 00011: val_acc did not improve from 0.74518\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.7730 - acc: 0.7324 - val_loss: 0.7632 - val_acc: 0.7282\n",
      "Epoch 12/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.7547 - acc: 0.7385\n",
      "Epoch 00012: val_acc improved from 0.74518 to 0.75051, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.12-0.75.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.7547 - acc: 0.7385 - val_loss: 0.6781 - val_acc: 0.7505\n",
      "Epoch 13/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.7344 - acc: 0.7376\n",
      "Epoch 00013: val_acc did not improve from 0.75051\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.7345 - acc: 0.7378 - val_loss: 0.6926 - val_acc: 0.7431\n",
      "Epoch 14/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.7093 - acc: 0.7492\n",
      "Epoch 00014: val_acc did not improve from 0.75051\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.7107 - acc: 0.7491 - val_loss: 0.7274 - val_acc: 0.7495\n",
      "Epoch 15/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.7078 - acc: 0.7504\n",
      "Epoch 00015: val_acc improved from 0.75051 to 0.77589, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.15-0.78.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.7078 - acc: 0.7504 - val_loss: 0.6469 - val_acc: 0.7759\n",
      "Epoch 16/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.6813 - acc: 0.7562\n",
      "Epoch 00016: val_acc improved from 0.77589 to 0.77716, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.16-0.78.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.6816 - acc: 0.7562 - val_loss: 0.6018 - val_acc: 0.7772\n",
      "Epoch 17/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.6597 - acc: 0.7603\n",
      "Epoch 00017: val_acc did not improve from 0.77716\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.6582 - acc: 0.7610 - val_loss: 0.6662 - val_acc: 0.7698\n",
      "Epoch 18/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.6715 - acc: 0.7607\n",
      "Epoch 00018: val_acc did not improve from 0.77716\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.6712 - acc: 0.7610 - val_loss: 0.6662 - val_acc: 0.7632\n",
      "Epoch 19/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.6402 - acc: 0.7720\n",
      "Epoch 00019: val_acc improved from 0.77716 to 0.79086, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.19-0.79.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.6400 - acc: 0.7720 - val_loss: 0.6031 - val_acc: 0.7909\n",
      "Epoch 20/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.6181 - acc: 0.7814\n",
      "Epoch 00020: val_acc improved from 0.79086 to 0.79162, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.20-0.79.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.6172 - acc: 0.7817 - val_loss: 0.5629 - val_acc: 0.7916\n",
      "Epoch 21/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.6143 - acc: 0.7805\n",
      "Epoch 00021: val_acc did not improve from 0.79162\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.6143 - acc: 0.7805 - val_loss: 0.6946 - val_acc: 0.7576\n",
      "Epoch 22/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.6026 - acc: 0.7856\n",
      "Epoch 00022: val_acc did not improve from 0.79162\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.6026 - acc: 0.7856 - val_loss: 0.9355 - val_acc: 0.6683\n",
      "Epoch 23/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5765 - acc: 0.7959\n",
      "Epoch 00023: val_acc improved from 0.79162 to 0.80558, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.23-0.81.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.5759 - acc: 0.7960 - val_loss: 0.5482 - val_acc: 0.8056\n",
      "Epoch 24/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5793 - acc: 0.7958\n",
      "Epoch 00024: val_acc did not improve from 0.80558\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.5784 - acc: 0.7958 - val_loss: 0.6193 - val_acc: 0.7695\n",
      "Epoch 25/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5698 - acc: 0.7982\n",
      "Epoch 00025: val_acc improved from 0.80558 to 0.80685, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.25-0.81.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.5690 - acc: 0.7986 - val_loss: 0.5302 - val_acc: 0.8069\n",
      "Epoch 26/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5574 - acc: 0.7975\n",
      "Epoch 00026: val_acc improved from 0.80685 to 0.81827, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.26-0.82.h5\n",
      "184/184 [==============================] - 9s 51ms/step - loss: 0.5574 - acc: 0.7976 - val_loss: 0.4906 - val_acc: 0.8183\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 27/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5415 - acc: 0.8098\n",
      "Epoch 00027: val_acc did not improve from 0.81827\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.5409 - acc: 0.8098 - val_loss: 0.6072 - val_acc: 0.7898\n",
      "Epoch 28/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5596 - acc: 0.8089\n",
      "Epoch 00028: val_acc did not improve from 0.81827\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.5587 - acc: 0.8091 - val_loss: 0.5411 - val_acc: 0.8043\n",
      "Epoch 29/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5534 - acc: 0.8081\n",
      "Epoch 00029: val_acc did not improve from 0.81827\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.5536 - acc: 0.8081 - val_loss: 0.6377 - val_acc: 0.7772\n",
      "Epoch 30/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5301 - acc: 0.8124\n",
      "Epoch 00030: val_acc improved from 0.81827 to 0.82868, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.30-0.83.h5\n",
      "184/184 [==============================] - 9s 51ms/step - loss: 0.5314 - acc: 0.8118 - val_loss: 0.4890 - val_acc: 0.8287\n",
      "Epoch 31/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4956 - acc: 0.8252\n",
      "Epoch 00031: val_acc did not improve from 0.82868\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.4972 - acc: 0.8243 - val_loss: 0.6029 - val_acc: 0.7855\n",
      "Epoch 32/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5143 - acc: 0.8207\n",
      "Epoch 00032: val_acc did not improve from 0.82868\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.5144 - acc: 0.8207 - val_loss: 0.5281 - val_acc: 0.8081\n",
      "Epoch 33/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.5171 - acc: 0.8192\n",
      "Epoch 00033: val_acc did not improve from 0.82868\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.5180 - acc: 0.8183 - val_loss: 0.5618 - val_acc: 0.8066\n",
      "Epoch 34/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.4887 - acc: 0.8295\n",
      "Epoch 00034: val_acc did not improve from 0.82868\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.4887 - acc: 0.8295 - val_loss: 0.5321 - val_acc: 0.8076\n",
      "Epoch 35/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4637 - acc: 0.8322\n",
      "Epoch 00035: val_acc did not improve from 0.82868\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.4634 - acc: 0.8324 - val_loss: 0.4914 - val_acc: 0.8183\n",
      "Epoch 36/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4590 - acc: 0.8390\n",
      "Epoch 00036: val_acc did not improve from 0.82868\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.4604 - acc: 0.8386 - val_loss: 0.4714 - val_acc: 0.8264\n",
      "Epoch 37/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4588 - acc: 0.8377\n",
      "Epoch 00037: val_acc improved from 0.82868 to 0.83122, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.37-0.83.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.4581 - acc: 0.8380 - val_loss: 0.5428 - val_acc: 0.8312\n",
      "Epoch 38/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4625 - acc: 0.8395\n",
      "Epoch 00038: val_acc did not improve from 0.83122\n",
      "184/184 [==============================] - 9s 51ms/step - loss: 0.4625 - acc: 0.8392 - val_loss: 0.5996 - val_acc: 0.7906\n",
      "Epoch 39/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4603 - acc: 0.8361\n",
      "Epoch 00039: val_acc improved from 0.83122 to 0.83198, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.39-0.83.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.4595 - acc: 0.8363 - val_loss: 0.4668 - val_acc: 0.8320\n",
      "Epoch 40/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4462 - acc: 0.8452\n",
      "Epoch 00040: val_acc improved from 0.83198 to 0.83503, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.40-0.84.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.4466 - acc: 0.8452 - val_loss: 0.4975 - val_acc: 0.8350\n",
      "Epoch 41/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.4407 - acc: 0.8440\n",
      "Epoch 00041: val_acc did not improve from 0.83503\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.4407 - acc: 0.8440 - val_loss: 0.6026 - val_acc: 0.7926\n",
      "Epoch 42/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4086 - acc: 0.8532\n",
      "Epoch 00042: val_acc improved from 0.83503 to 0.83731, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.42-0.84.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.4071 - acc: 0.8540 - val_loss: 0.4503 - val_acc: 0.8373\n",
      "Epoch 43/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.4222 - acc: 0.8525\n",
      "Epoch 00043: val_acc did not improve from 0.83731\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.4222 - acc: 0.8525 - val_loss: 0.5202 - val_acc: 0.8145\n",
      "Epoch 44/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3998 - acc: 0.8616\n",
      "Epoch 00044: val_acc improved from 0.83731 to 0.84594, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.44-0.85.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.4013 - acc: 0.8613 - val_loss: 0.4305 - val_acc: 0.8459\n",
      "Epoch 45/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3979 - acc: 0.8599\n",
      "Epoch 00045: val_acc did not improve from 0.84594\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.3991 - acc: 0.8596 - val_loss: 0.5100 - val_acc: 0.8160\n",
      "Epoch 46/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.4088 - acc: 0.8567\n",
      "Epoch 00046: val_acc did not improve from 0.84594\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.4087 - acc: 0.8566 - val_loss: 0.4379 - val_acc: 0.8449\n",
      "Epoch 47/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3950 - acc: 0.8606\n",
      "Epoch 00047: val_acc improved from 0.84594 to 0.84695, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.47-0.85.h5\n",
      "184/184 [==============================] - 10s 52ms/step - loss: 0.3943 - acc: 0.8608 - val_loss: 0.4265 - val_acc: 0.8470\n",
      "Epoch 48/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3976 - acc: 0.8592\n",
      "Epoch 00048: val_acc improved from 0.84695 to 0.85025, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.48-0.85.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.3979 - acc: 0.8590 - val_loss: 0.4369 - val_acc: 0.8503\n",
      "Epoch 49/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3861 - acc: 0.8633\n",
      "Epoch 00049: val_acc did not improve from 0.85025\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.3868 - acc: 0.8630 - val_loss: 0.5141 - val_acc: 0.8178\n",
      "Epoch 50/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.3936 - acc: 0.8629\n",
      "Epoch 00050: val_acc did not improve from 0.85025\n",
      "184/184 [==============================] - 10s 56ms/step - loss: 0.3936 - acc: 0.8629 - val_loss: 0.4994 - val_acc: 0.8208\n",
      "lr changed to 0.0009999999776482583\n",
      "Epoch 51/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3635 - acc: 0.8715\n",
      "Epoch 00051: val_acc improved from 0.85025 to 0.86091, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.51-0.86.h5\n",
      "184/184 [==============================] - 10s 54ms/step - loss: 0.3637 - acc: 0.8712 - val_loss: 0.3881 - val_acc: 0.8609\n",
      "Epoch 52/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.3372 - acc: 0.8826\n",
      "Epoch 00052: val_acc did not improve from 0.86091\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.3372 - acc: 0.8826 - val_loss: 0.4009 - val_acc: 0.8581\n",
      "Epoch 53/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3266 - acc: 0.8888\n",
      "Epoch 00053: val_acc did not improve from 0.86091\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.3273 - acc: 0.8887 - val_loss: 0.3968 - val_acc: 0.8579\n",
      "Epoch 54/200\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "183/184 [============================>.] - ETA: 0s - loss: 0.3399 - acc: 0.8832\n",
      "Epoch 00054: val_acc did not improve from 0.86091\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.3400 - acc: 0.8831 - val_loss: 0.4069 - val_acc: 0.8551\n",
      "Epoch 55/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3262 - acc: 0.8857\n",
      "Epoch 00055: val_acc did not improve from 0.86091\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.3258 - acc: 0.8860 - val_loss: 0.3891 - val_acc: 0.8596\n",
      "Epoch 56/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3242 - acc: 0.8842\n",
      "Epoch 00056: val_acc did not improve from 0.86091\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.3243 - acc: 0.8843 - val_loss: 0.4051 - val_acc: 0.8541\n",
      "Epoch 57/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3099 - acc: 0.8876\n",
      "Epoch 00057: val_acc improved from 0.86091 to 0.86142, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.57-0.86.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.3101 - acc: 0.8875 - val_loss: 0.3837 - val_acc: 0.8614\n",
      "Epoch 58/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3247 - acc: 0.8849\n",
      "Epoch 00058: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.3244 - acc: 0.8852 - val_loss: 0.4001 - val_acc: 0.8558\n",
      "Epoch 59/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3170 - acc: 0.8895\n",
      "Epoch 00059: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 52ms/step - loss: 0.3178 - acc: 0.8892 - val_loss: 0.3868 - val_acc: 0.8581\n",
      "Epoch 60/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3086 - acc: 0.8902\n",
      "Epoch 00060: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 51ms/step - loss: 0.3091 - acc: 0.8901 - val_loss: 0.3904 - val_acc: 0.8589\n",
      "Epoch 61/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3075 - acc: 0.8912\n",
      "Epoch 00061: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.3071 - acc: 0.8913 - val_loss: 0.3985 - val_acc: 0.8566\n",
      "Epoch 62/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3022 - acc: 0.8946\n",
      "Epoch 00062: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.3034 - acc: 0.8940 - val_loss: 0.4029 - val_acc: 0.8563\n",
      "Epoch 63/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2985 - acc: 0.8991\n",
      "Epoch 00063: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2991 - acc: 0.8986 - val_loss: 0.4027 - val_acc: 0.8563\n",
      "Epoch 64/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3042 - acc: 0.8933\n",
      "Epoch 00064: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.3038 - acc: 0.8933 - val_loss: 0.4091 - val_acc: 0.8528\n",
      "Epoch 65/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3095 - acc: 0.8927\n",
      "Epoch 00065: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.3087 - acc: 0.8932 - val_loss: 0.3935 - val_acc: 0.8574\n",
      "Epoch 66/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2912 - acc: 0.8951\n",
      "Epoch 00066: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2910 - acc: 0.8954 - val_loss: 0.3973 - val_acc: 0.8596\n",
      "Epoch 67/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3042 - acc: 0.8929\n",
      "Epoch 00067: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.3040 - acc: 0.8928 - val_loss: 0.4040 - val_acc: 0.8569\n",
      "Epoch 68/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3088 - acc: 0.8955\n",
      "Epoch 00068: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.3086 - acc: 0.8955 - val_loss: 0.3968 - val_acc: 0.8604\n",
      "Epoch 69/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2935 - acc: 0.8934\n",
      "Epoch 00069: val_acc did not improve from 0.86142\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2928 - acc: 0.8937 - val_loss: 0.3955 - val_acc: 0.8596\n",
      "Epoch 70/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3041 - acc: 0.8922\n",
      "Epoch 00070: val_acc improved from 0.86142 to 0.86193, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.70-0.86.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.3033 - acc: 0.8925 - val_loss: 0.3833 - val_acc: 0.8619\n",
      "Epoch 71/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2926 - acc: 0.8950\n",
      "Epoch 00071: val_acc improved from 0.86193 to 0.86320, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.71-0.86.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2928 - acc: 0.8947 - val_loss: 0.3882 - val_acc: 0.8632\n",
      "Epoch 72/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3006 - acc: 0.8933\n",
      "Epoch 00072: val_acc improved from 0.86320 to 0.86345, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.72-0.86.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.3016 - acc: 0.8930 - val_loss: 0.3796 - val_acc: 0.8635\n",
      "Epoch 73/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.3014 - acc: 0.8943\n",
      "Epoch 00073: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.3012 - acc: 0.8944 - val_loss: 0.3996 - val_acc: 0.8543\n",
      "Epoch 74/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2952 - acc: 0.8963\n",
      "Epoch 00074: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2960 - acc: 0.8962 - val_loss: 0.4053 - val_acc: 0.8515\n",
      "Epoch 75/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2858 - acc: 0.9027\n",
      "Epoch 00075: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2865 - acc: 0.9022 - val_loss: 0.3932 - val_acc: 0.8576\n",
      "Epoch 76/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2904 - acc: 0.8987\n",
      "Epoch 00076: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2897 - acc: 0.8991 - val_loss: 0.3970 - val_acc: 0.8536\n",
      "Epoch 77/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2873 - acc: 0.9003\n",
      "Epoch 00077: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2871 - acc: 0.9003 - val_loss: 0.3959 - val_acc: 0.8569\n",
      "Epoch 78/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2818 - acc: 0.8979\n",
      "Epoch 00078: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2821 - acc: 0.8979 - val_loss: 0.3966 - val_acc: 0.8569\n",
      "Epoch 79/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2935 - acc: 0.8969\n",
      "Epoch 00079: val_acc did not improve from 0.86345\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2937 - acc: 0.8971 - val_loss: 0.3918 - val_acc: 0.8629\n",
      "Epoch 80/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2881 - acc: 0.9052\n",
      "Epoch 00080: val_acc improved from 0.86345 to 0.86624, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.80-0.87.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2885 - acc: 0.9047 - val_loss: 0.3842 - val_acc: 0.8662\n",
      "Epoch 81/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2904 - acc: 0.8982\n",
      "Epoch 00081: val_acc did not improve from 0.86624\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2910 - acc: 0.8979 - val_loss: 0.4109 - val_acc: 0.8541\n",
      "Epoch 82/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2827 - acc: 0.8998\n",
      "Epoch 00082: val_acc did not improve from 0.86624\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2849 - acc: 0.8989 - val_loss: 0.3927 - val_acc: 0.8640\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 83/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2827 - acc: 0.9003\n",
      "Epoch 00083: val_acc did not improve from 0.86624\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2824 - acc: 0.9005 - val_loss: 0.3872 - val_acc: 0.8652\n",
      "Epoch 84/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2798 - acc: 0.9003\n",
      "Epoch 00084: val_acc did not improve from 0.86624\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2798 - acc: 0.9003 - val_loss: 0.3847 - val_acc: 0.8655\n",
      "Epoch 85/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2767 - acc: 0.9025\n",
      "Epoch 00085: val_acc improved from 0.86624 to 0.86751, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.85-0.87.h5\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.2772 - acc: 0.9027 - val_loss: 0.3775 - val_acc: 0.8675\n",
      "Epoch 86/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2824 - acc: 0.8932\n",
      "Epoch 00086: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 51ms/step - loss: 0.2824 - acc: 0.8932 - val_loss: 0.3867 - val_acc: 0.8614\n",
      "Epoch 87/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2778 - acc: 0.9030\n",
      "Epoch 00087: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2778 - acc: 0.9030 - val_loss: 0.3840 - val_acc: 0.8607\n",
      "Epoch 88/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2812 - acc: 0.9025\n",
      "Epoch 00088: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2812 - acc: 0.9025 - val_loss: 0.3937 - val_acc: 0.8614\n",
      "Epoch 89/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2794 - acc: 0.8986\n",
      "Epoch 00089: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2795 - acc: 0.8984 - val_loss: 0.3767 - val_acc: 0.8645\n",
      "Epoch 90/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2847 - acc: 0.9003\n",
      "Epoch 00090: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2843 - acc: 0.9003 - val_loss: 0.3799 - val_acc: 0.8614\n",
      "Epoch 91/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2697 - acc: 0.9059\n",
      "Epoch 00091: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2698 - acc: 0.9056 - val_loss: 0.3914 - val_acc: 0.8614\n",
      "Epoch 92/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2834 - acc: 0.9025\n",
      "Epoch 00092: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2835 - acc: 0.9023 - val_loss: 0.3776 - val_acc: 0.8640\n",
      "Epoch 93/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2754 - acc: 0.9040\n",
      "Epoch 00093: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2766 - acc: 0.9037 - val_loss: 0.3829 - val_acc: 0.8619\n",
      "Epoch 94/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2713 - acc: 0.9044\n",
      "Epoch 00094: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2710 - acc: 0.9042 - val_loss: 0.3867 - val_acc: 0.8602\n",
      "Epoch 95/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2748 - acc: 0.9013\n",
      "Epoch 00095: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2748 - acc: 0.9012 - val_loss: 0.3881 - val_acc: 0.8614\n",
      "Epoch 96/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2736 - acc: 0.9022\n",
      "Epoch 00096: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2736 - acc: 0.9022 - val_loss: 0.3912 - val_acc: 0.8602\n",
      "Epoch 97/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2753 - acc: 0.9063\n",
      "Epoch 00097: val_acc did not improve from 0.86751\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2753 - acc: 0.9063 - val_loss: 0.3826 - val_acc: 0.8635\n",
      "Epoch 98/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2826 - acc: 0.8971\n",
      "Epoch 00098: val_acc improved from 0.86751 to 0.86777, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.98-0.87.h5\n",
      "184/184 [==============================] - 9s 51ms/step - loss: 0.2826 - acc: 0.8971 - val_loss: 0.3837 - val_acc: 0.8678\n",
      "Epoch 99/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2731 - acc: 0.9073\n",
      "Epoch 00099: val_acc did not improve from 0.86777\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2732 - acc: 0.9073 - val_loss: 0.3812 - val_acc: 0.8670\n",
      "Epoch 100/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2753 - acc: 0.9006\n",
      "Epoch 00100: val_acc did not improve from 0.86777\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2748 - acc: 0.9010 - val_loss: 0.3839 - val_acc: 0.8627\n",
      "lr changed to 9.999999310821295e-05\n",
      "Epoch 101/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2641 - acc: 0.9068\n",
      "Epoch 00101: val_acc did not improve from 0.86777\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2641 - acc: 0.9068 - val_loss: 0.3807 - val_acc: 0.8662\n",
      "Epoch 102/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2753 - acc: 0.9013\n",
      "Epoch 00102: val_acc did not improve from 0.86777\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2753 - acc: 0.9012 - val_loss: 0.3801 - val_acc: 0.8660\n",
      "Epoch 103/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2600 - acc: 0.9056\n",
      "Epoch 00103: val_acc improved from 0.86777 to 0.86802, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.103-0.87.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2603 - acc: 0.9056 - val_loss: 0.3779 - val_acc: 0.8680\n",
      "Epoch 104/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2730 - acc: 0.9046\n",
      "Epoch 00104: val_acc did not improve from 0.86802\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2733 - acc: 0.9046 - val_loss: 0.3786 - val_acc: 0.8660\n",
      "Epoch 105/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2688 - acc: 0.9030\n",
      "Epoch 00105: val_acc did not improve from 0.86802\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2688 - acc: 0.9030 - val_loss: 0.3794 - val_acc: 0.8665\n",
      "Epoch 106/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2602 - acc: 0.9093\n",
      "Epoch 00106: val_acc improved from 0.86802 to 0.86853, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.106-0.87.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2600 - acc: 0.9097 - val_loss: 0.3765 - val_acc: 0.8685\n",
      "Epoch 107/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2675 - acc: 0.9073\n",
      "Epoch 00107: val_acc did not improve from 0.86853\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2676 - acc: 0.9073 - val_loss: 0.3774 - val_acc: 0.8665\n",
      "Epoch 108/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2658 - acc: 0.9063\n",
      "Epoch 00108: val_acc did not improve from 0.86853\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2655 - acc: 0.9066 - val_loss: 0.3763 - val_acc: 0.8678\n",
      "Epoch 109/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2626 - acc: 0.9069\n",
      "Epoch 00109: val_acc did not improve from 0.86853\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2629 - acc: 0.9066 - val_loss: 0.3758 - val_acc: 0.8678\n",
      "Epoch 110/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2714 - acc: 0.9076\n",
      "Epoch 00110: val_acc did not improve from 0.86853\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2709 - acc: 0.9075 - val_loss: 0.3770 - val_acc: 0.8680\n",
      "Epoch 111/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2689 - acc: 0.9037\n",
      "Epoch 00111: val_acc did not improve from 0.86853\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2687 - acc: 0.9037 - val_loss: 0.3781 - val_acc: 0.8670\n",
      "Epoch 112/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2614 - acc: 0.9075\n",
      "Epoch 00112: val_acc did not improve from 0.86853\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2614 - acc: 0.9071 - val_loss: 0.3773 - val_acc: 0.8665\n",
      "Epoch 113/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2641 - acc: 0.9100\n",
      "Epoch 00113: val_acc improved from 0.86853 to 0.86878, saving model to D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data/trained_model\\resnet50_model.113-0.87.h5\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2641 - acc: 0.9100 - val_loss: 0.3753 - val_acc: 0.8688\n",
      "Epoch 114/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2554 - acc: 0.9075\n",
      "Epoch 00114: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2560 - acc: 0.9073 - val_loss: 0.3781 - val_acc: 0.8657\n",
      "Epoch 115/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2695 - acc: 0.9064\n",
      "Epoch 00115: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2694 - acc: 0.9064 - val_loss: 0.3792 - val_acc: 0.8652\n",
      "Epoch 116/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2674 - acc: 0.9051\n",
      "Epoch 00116: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.2667 - acc: 0.9054 - val_loss: 0.3786 - val_acc: 0.8660\n",
      "Epoch 117/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2762 - acc: 0.9064\n",
      "Epoch 00117: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.2762 - acc: 0.9064 - val_loss: 0.3782 - val_acc: 0.8647\n",
      "Epoch 118/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2681 - acc: 0.9056\n",
      "Epoch 00118: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2674 - acc: 0.9058 - val_loss: 0.3795 - val_acc: 0.8640\n",
      "Epoch 119/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2629 - acc: 0.9097\n",
      "Epoch 00119: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2629 - acc: 0.9097 - val_loss: 0.3812 - val_acc: 0.8655\n",
      "Epoch 120/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2631 - acc: 0.9097\n",
      "Epoch 00120: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2629 - acc: 0.9097 - val_loss: 0.3794 - val_acc: 0.8655\n",
      "Epoch 121/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2600 - acc: 0.9085\n",
      "Epoch 00121: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2611 - acc: 0.9081 - val_loss: 0.3812 - val_acc: 0.8650\n",
      "Epoch 122/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2579 - acc: 0.9116\n",
      "Epoch 00122: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2573 - acc: 0.9117 - val_loss: 0.3813 - val_acc: 0.8657\n",
      "Epoch 123/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2591 - acc: 0.9087\n",
      "Epoch 00123: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2584 - acc: 0.9092 - val_loss: 0.3806 - val_acc: 0.8660\n",
      "Epoch 124/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2560 - acc: 0.9085\n",
      "Epoch 00124: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2560 - acc: 0.9085 - val_loss: 0.3788 - val_acc: 0.8668\n",
      "Epoch 125/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2618 - acc: 0.9061\n",
      "Epoch 00125: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2621 - acc: 0.9063 - val_loss: 0.3799 - val_acc: 0.8657\n",
      "Epoch 126/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2592 - acc: 0.9076\n",
      "Epoch 00126: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 48ms/step - loss: 0.2592 - acc: 0.9075 - val_loss: 0.3799 - val_acc: 0.8647\n",
      "Epoch 127/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2542 - acc: 0.9099\n",
      "Epoch 00127: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2546 - acc: 0.9098 - val_loss: 0.3804 - val_acc: 0.8660\n",
      "Epoch 128/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2531 - acc: 0.9095\n",
      "Epoch 00128: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2538 - acc: 0.9093 - val_loss: 0.3799 - val_acc: 0.8647\n",
      "Epoch 129/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2751 - acc: 0.9035\n",
      "Epoch 00129: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 50ms/step - loss: 0.2750 - acc: 0.9035 - val_loss: 0.3805 - val_acc: 0.8662\n",
      "Epoch 130/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2725 - acc: 0.9063\n",
      "Epoch 00130: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2724 - acc: 0.9061 - val_loss: 0.3813 - val_acc: 0.8647\n",
      "Epoch 131/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2591 - acc: 0.9086\n",
      "Epoch 00131: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2591 - acc: 0.9086 - val_loss: 0.3808 - val_acc: 0.8650\n",
      "Epoch 132/200\n",
      "184/184 [==============================] - ETA: 0s - loss: 0.2640 - acc: 0.9037\n",
      "Epoch 00132: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 49ms/step - loss: 0.2640 - acc: 0.9037 - val_loss: 0.3817 - val_acc: 0.8665\n",
      "Epoch 133/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2608 - acc: 0.9080\n",
      "Epoch 00133: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2621 - acc: 0.9073 - val_loss: 0.3804 - val_acc: 0.8668\n",
      "Epoch 134/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2638 - acc: 0.9052\n",
      "Epoch 00134: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2633 - acc: 0.9052 - val_loss: 0.3803 - val_acc: 0.8662\n",
      "Epoch 135/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2576 - acc: 0.9116\n",
      "Epoch 00135: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2578 - acc: 0.9114 - val_loss: 0.3785 - val_acc: 0.8675\n",
      "Epoch 136/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2693 - acc: 0.9061\n",
      "Epoch 00136: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2687 - acc: 0.9059 - val_loss: 0.3787 - val_acc: 0.8685\n",
      "Epoch 137/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2660 - acc: 0.9069\n",
      "Epoch 00137: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2654 - acc: 0.9073 - val_loss: 0.3777 - val_acc: 0.8678\n",
      "Epoch 138/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2730 - acc: 0.9064\n",
      "Epoch 00138: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2722 - acc: 0.9068 - val_loss: 0.3794 - val_acc: 0.8665\n",
      "Epoch 139/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2671 - acc: 0.9063\n",
      "Epoch 00139: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2673 - acc: 0.9063 - val_loss: 0.3797 - val_acc: 0.8655\n",
      "Epoch 140/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2582 - acc: 0.9090\n",
      "Epoch 00140: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 8s 46ms/step - loss: 0.2574 - acc: 0.9095 - val_loss: 0.3792 - val_acc: 0.8665\n",
      "Epoch 141/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2675 - acc: 0.9052\n",
      "Epoch 00141: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2677 - acc: 0.9051 - val_loss: 0.3779 - val_acc: 0.8673\n",
      "Epoch 142/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2586 - acc: 0.9075\n",
      "Epoch 00142: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 46ms/step - loss: 0.2581 - acc: 0.9075 - val_loss: 0.3778 - val_acc: 0.8678\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch 143/200\n",
      "183/184 [============================>.] - ETA: 0s - loss: 0.2588 - acc: 0.9099Restoring model weights from the end of the best epoch.\n",
      "\n",
      "Epoch 00143: val_acc did not improve from 0.86878\n",
      "184/184 [==============================] - 9s 47ms/step - loss: 0.2583 - acc: 0.9098 - val_loss: 0.3809 - val_acc: 0.8665\n",
      "Epoch 00143: early stopping\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "<tensorflow.python.keras.callbacks.History at 0x1f2a7e6f400>"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "batch_size = 32\n",
    "epochs = 200\n",
    "iterations = 184\n",
    "ResNet_model.fit_generator(train_data_gen.flow(train_x, train_y, batch_size=batch_size),\n",
    "                    steps_per_epoch=iterations,\n",
    "                    epochs=epochs,\n",
    "                    callbacks=cbks,\n",
    "                    validation_data=valid_data_gen.flow(test_x, test_y, batch_size=batch_size))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 71,
   "metadata": {},
   "outputs": [],
   "source": [
    "from tensorflow.keras.models import model_from_json\n",
    "json_file = open(\"./data/neural_networks/Resnet_model.json\", 'r')\n",
    "loaded_model_json = json_file.read()\n",
    "json_file.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "ename": "NameError",
     "evalue": "name 'model' is not defined",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[1;31mNameError\u001b[0m                                 Traceback (most recent call last)",
      "\u001b[1;32m<ipython-input-8-85e4a7ce47d6>\u001b[0m in \u001b[0;36m<module>\u001b[1;34m\u001b[0m\n\u001b[1;32m----> 1\u001b[1;33m \u001b[0mmodel\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0msave_weights\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;34m'model.h5'\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m",
      "\u001b[1;31mNameError\u001b[0m: name 'model' is not defined"
     ]
    }
   ],
   "source": [
    "model.save_weights('model.h5')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_perdict = model(train_x)\n",
    "train_perdict_arg = np.argmax(train_perdict, axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([1, 1, 2, ..., 1, 5, 5], dtype=int64)"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_perdict_arg"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_y_arg = np.argmax(train_y, axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([1, 1, 2, ..., 1, 5, 5], dtype=int64)"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_y_arg"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.metrics import classification_report\n",
    "report = classification_report(train_y_arg,train_perdict_arg)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       0.78      0.78      0.78       820\n",
      "           1       0.95      0.96      0.95      3458\n",
      "           2       0.79      0.69      0.74       529\n",
      "           3       0.72      0.78      0.75       279\n",
      "           4       0.98      0.92      0.95       326\n",
      "           5       0.92      0.96      0.94       498\n",
      "\n",
      "    accuracy                           0.90      5910\n",
      "   macro avg       0.86      0.85      0.85      5910\n",
      "weighted avg       0.90      0.90      0.90      5910\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(report)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 110,
   "metadata": {},
   "outputs": [],
   "source": [
    "GF_model = model_from_json(loaded_model_json)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 80,
   "metadata": {},
   "outputs": [],
   "source": [
    "filepath = 'D:\\Projects\\PycharmProjects\\Grad\\BITVehicle\\data\\neural_networks\\GTSRB_VGG19_model_00_ochiai_GF.h5'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 105,
   "metadata": {},
   "outputs": [],
   "source": [
    "GF_model.load_weights('model3.h5')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 106,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_perdict = GF_model(test_x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 135,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_perdict_arg = np.argmax(test_perdict, axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 134,
   "metadata": {},
   "outputs": [],
   "source": [
    "GF_model.load_weights('BIT_RESNET50_model_02_tarantula_GF_44.h5')\n",
    "test_perdict = GF_model(test_x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 76,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[array([[[[-3.98619995e-02, -3.22993249e-02,  1.51274092e-02, ...,\n",
       "            2.32277699e-02,  1.55955553e-04, -2.81602684e-02],\n",
       "          [ 1.92677379e-02,  1.45063810e-02, -4.20132577e-02, ...,\n",
       "            3.69417630e-02, -4.23839577e-02,  1.25940554e-02],\n",
       "          [ 2.33327411e-02, -4.23488356e-02, -3.26490216e-02, ...,\n",
       "           -3.62713486e-02,  1.27105527e-02,  2.17687301e-02]],\n",
       " \n",
       "         [[ 3.45859118e-02, -4.07557599e-02, -3.06687020e-02, ...,\n",
       "            2.22389214e-02,  1.26944296e-02, -3.31287496e-02],\n",
       "          [ 2.86565535e-02, -3.26312557e-02,  1.09715536e-02, ...,\n",
       "           -4.17886376e-02,  1.25044622e-02,  1.01410598e-02],\n",
       "          [ 1.92729160e-02,  1.32944621e-02,  3.52579989e-02, ...,\n",
       "            8.24379921e-03,  1.68838985e-02, -2.87031103e-02]],\n",
       " \n",
       "         [[-3.34351286e-02, -3.18679810e-02,  1.22405477e-02, ...,\n",
       "            1.91206932e-02, -2.94669662e-02, -5.76304644e-03],\n",
       "          [ 3.94372381e-02, -8.59326869e-03, -4.20898944e-02, ...,\n",
       "            1.41411833e-02,  2.84244604e-02, -1.86693072e-02],\n",
       "          [ 2.10816525e-02,  3.02948989e-02, -1.32805891e-02, ...,\n",
       "            5.74329495e-03, -3.83874215e-02, -3.85779217e-02]],\n",
       " \n",
       "         ...,\n",
       " \n",
       "         [[-2.23863553e-02, -3.35706249e-02, -1.09419823e-02, ...,\n",
       "           -2.56049633e-03, -2.44766511e-02, -2.43559834e-02],\n",
       "          [-5.93661517e-03, -2.90849097e-02,  1.32656172e-02, ...,\n",
       "           -2.30067633e-02, -2.05899402e-03, -3.00589353e-02],\n",
       "          [-1.44224949e-02, -2.03781594e-02, -3.87417153e-02, ...,\n",
       "            8.81474093e-03, -9.94170457e-03, -1.21327005e-02]],\n",
       " \n",
       "         [[ 6.50947541e-03, -3.18368226e-02, -1.22460313e-02, ...,\n",
       "           -2.06768885e-03,  1.96656324e-02, -2.63110138e-02],\n",
       "          [-3.77444290e-02,  2.49406956e-02, -3.23057063e-02, ...,\n",
       "           -1.37289371e-02, -1.05287619e-02, -3.18659544e-02],\n",
       "          [-8.85624439e-03,  8.13684613e-03,  6.21598214e-04, ...,\n",
       "           -7.76758417e-03,  2.78707333e-02, -6.25753775e-03]],\n",
       " \n",
       "         [[ 3.56800668e-02,  3.61658633e-04, -1.45297609e-02, ...,\n",
       "            1.69668682e-02,  1.28206238e-03,  2.75585167e-02],\n",
       "          [-3.27276960e-02,  1.08288080e-02, -8.97325575e-04, ...,\n",
       "           -1.56071484e-02, -3.73244882e-02, -4.02055085e-02],\n",
       "          [ 3.55897956e-02,  8.47597420e-03, -3.75495404e-02, ...,\n",
       "           -3.73670310e-02, -1.64678730e-02, -1.12132356e-02]]],\n",
       " \n",
       " \n",
       "        [[[-6.28128648e-03,  3.75373773e-02, -3.72217596e-02, ...,\n",
       "            1.65480860e-02, -3.60611193e-02, -5.50569594e-03],\n",
       "          [-1.05684809e-02, -2.10110229e-02,  6.67871162e-03, ...,\n",
       "           -1.13134086e-02,  1.47895180e-02, -1.43291522e-02],\n",
       "          [ 2.03164481e-02,  9.03436914e-03,  1.39068775e-02, ...,\n",
       "            2.34462358e-02,  2.10766084e-02,  2.27386467e-02]],\n",
       " \n",
       "         [[-3.48723494e-02, -4.07046080e-03,  3.65557894e-03, ...,\n",
       "            3.00288983e-02, -3.41112874e-02,  9.72466543e-03],\n",
       "          [-4.07919623e-02,  1.68618634e-02,  9.72528756e-03, ...,\n",
       "           -1.47823412e-02,  1.87417977e-02, -1.95961203e-02],\n",
       "          [ 2.09606327e-02, -4.00090888e-02, -6.76348060e-03, ...,\n",
       "            1.11763701e-02, -3.07241678e-02,  1.00865401e-02]],\n",
       " \n",
       "         [[ 1.75116435e-02,  2.98248045e-02,  4.18870263e-02, ...,\n",
       "            5.65269217e-03, -2.02979557e-02, -2.52033658e-02],\n",
       "          [-5.08574769e-03, -1.38670951e-02, -9.62257758e-03, ...,\n",
       "            1.86020881e-02,  3.29063945e-02,  6.71594217e-03],\n",
       "          [-2.79539730e-02, -1.43298563e-02, -1.66689809e-02, ...,\n",
       "            1.84611455e-02,  3.67675722e-03, -3.25151309e-02]],\n",
       " \n",
       "         ...,\n",
       " \n",
       "         [[-2.00828221e-02,  3.39019857e-02,  2.08299123e-02, ...,\n",
       "           -2.35446710e-02, -1.47486236e-02, -4.35052067e-03],\n",
       "          [ 7.53698871e-03, -1.86087340e-02, -3.27788331e-02, ...,\n",
       "            1.20229907e-02,  4.02851589e-02, -1.07655227e-02],\n",
       "          [ 1.29654668e-02, -3.38756368e-02,  4.23251800e-02, ...,\n",
       "            1.98858604e-03,  1.68771297e-03,  1.94578096e-02]],\n",
       " \n",
       "         [[ 5.15229255e-03,  1.03919879e-02, -1.99694727e-02, ...,\n",
       "            1.67061016e-03,  4.11360897e-02, -3.59872356e-03],\n",
       "          [-9.43852216e-03,  1.01747327e-02, -2.80541852e-02, ...,\n",
       "           -3.17599997e-03,  2.42594071e-02, -3.14941294e-02],\n",
       "          [-4.63801995e-03,  2.50266492e-03,  2.60567926e-02, ...,\n",
       "            1.15554407e-03,  9.58310068e-03, -2.03787107e-02]],\n",
       " \n",
       "         [[-3.89759690e-02, -3.44047397e-02, -2.55696531e-02, ...,\n",
       "           -5.61007857e-03,  4.00645323e-02,  7.55982101e-03],\n",
       "          [-2.39830408e-02,  1.29919387e-02, -3.31647508e-02, ...,\n",
       "           -1.20637789e-02, -8.05901736e-03, -2.50206348e-02],\n",
       "          [-2.08168756e-02,  1.41193382e-02, -1.78089496e-02, ...,\n",
       "           -1.44220516e-03, -3.53194214e-02, -8.18205997e-03]]],\n",
       " \n",
       " \n",
       "        [[[ 8.61721113e-03, -3.09392512e-02,  2.83915922e-03, ...,\n",
       "           -1.62708014e-02,  6.22291118e-04,  5.06603345e-03],\n",
       "          [ 2.13476978e-02, -2.27763820e-02,  4.12082933e-02, ...,\n",
       "            2.57891156e-02,  2.78007053e-02,  1.55661553e-02],\n",
       "          [-2.99450643e-02, -1.95077937e-02,  1.46750547e-02, ...,\n",
       "           -2.43267044e-03, -3.59250605e-03,  1.98138617e-02]],\n",
       " \n",
       "         [[ 2.38899328e-02, -1.30443685e-02, -2.84200646e-02, ...,\n",
       "           -2.56468821e-02, -3.04868575e-02,  1.49088614e-02],\n",
       "          [-2.77254060e-02, -8.22982192e-03, -2.65894737e-02, ...,\n",
       "           -9.43315774e-03, -3.56293954e-02, -4.23632190e-02],\n",
       "          [ 9.60154831e-03, -1.43563971e-02, -2.60143522e-02, ...,\n",
       "            3.29566486e-02,  4.15311381e-03, -2.30652094e-03]],\n",
       " \n",
       "         [[-2.96595432e-02,  3.75710763e-02, -2.14708969e-03, ...,\n",
       "           -4.99968231e-03, -4.27004471e-02, -7.33175501e-03],\n",
       "          [ 1.99261568e-02,  1.93949118e-02, -4.25096937e-02, ...,\n",
       "            9.22496617e-03,  4.97648120e-03,  2.83067562e-02],\n",
       "          [-1.13311633e-02, -4.16499488e-02, -1.05500594e-03, ...,\n",
       "           -2.98165902e-03,  3.10845301e-03,  2.05519386e-02]],\n",
       " \n",
       "         ...,\n",
       " \n",
       "         [[-7.87347555e-03,  2.17451788e-02, -3.11173499e-05, ...,\n",
       "            1.35103166e-02,  1.88282877e-03, -3.11893541e-02],\n",
       "          [-2.92455852e-02, -2.66888365e-03,  4.10972051e-02, ...,\n",
       "           -3.05485427e-02, -6.78993762e-04, -2.23284923e-02],\n",
       "          [ 2.01433934e-02,  2.54786722e-02, -2.05878299e-02, ...,\n",
       "            6.17638230e-03,  3.31957974e-02, -1.71664655e-02]],\n",
       " \n",
       "         [[ 2.59818099e-02,  1.75457783e-02,  3.34009714e-02, ...,\n",
       "           -3.62450629e-02, -3.90305072e-02, -1.47756562e-03],\n",
       "          [ 4.01949920e-02,  2.45593973e-02, -3.67829502e-02, ...,\n",
       "           -1.21937040e-02, -2.75765248e-02, -1.63927134e-02],\n",
       "          [-1.63839273e-02,  2.41733007e-02, -2.84382887e-02, ...,\n",
       "           -2.10732166e-02, -3.35988700e-02,  5.25329262e-04]],\n",
       " \n",
       "         [[-3.76078300e-02, -2.28707343e-02,  2.05492973e-03, ...,\n",
       "           -4.14355174e-02, -1.06307864e-02, -2.82080919e-02],\n",
       "          [-1.21341273e-02,  3.62389497e-02,  2.44075432e-03, ...,\n",
       "           -3.29765454e-02, -3.82308885e-02,  8.97234678e-03],\n",
       "          [ 2.83511542e-02,  4.59818915e-03, -2.31320281e-02, ...,\n",
       "            2.18180977e-02, -9.27244499e-03,  1.71254724e-02]]],\n",
       " \n",
       " \n",
       "        ...,\n",
       " \n",
       " \n",
       "        [[[ 2.16808543e-03, -2.57308073e-02,  1.89179331e-02, ...,\n",
       "            4.22793441e-02,  1.59855038e-02, -2.47432664e-02],\n",
       "          [ 3.21229585e-02,  3.89737971e-02,  2.89303996e-02, ...,\n",
       "           -1.57908984e-02, -1.01665705e-02, -6.17611781e-03],\n",
       "          [ 1.20221637e-02,  2.67055966e-02, -2.77140513e-02, ...,\n",
       "            2.55604088e-03, -2.88093369e-02, -3.82996760e-02]],\n",
       " \n",
       "         [[-2.68818866e-02,  4.25125994e-02, -2.83672884e-02, ...,\n",
       "            3.22593115e-02,  4.04627137e-02, -7.44692981e-04],\n",
       "          [ 1.03053004e-02, -1.92943923e-02,  8.97258148e-03, ...,\n",
       "            2.82745548e-02,  2.50132494e-02,  2.06506997e-03],\n",
       "          [-1.02111623e-02, -3.77043635e-02, -2.35030148e-02, ...,\n",
       "           -4.16928977e-02,  2.57645436e-02, -3.82899530e-02]],\n",
       " \n",
       "         [[ 8.66311789e-03,  3.65271382e-02,  3.89070846e-02, ...,\n",
       "           -1.23715624e-02, -1.49545427e-02, -7.01288134e-03],\n",
       "          [-2.00502984e-02,  1.67972222e-02,  1.80959404e-02, ...,\n",
       "            2.71122493e-02,  1.96034275e-02,  8.22678208e-04],\n",
       "          [-2.32538786e-02,  5.15672565e-03,  2.77661346e-02, ...,\n",
       "           -3.72305028e-02,  2.39178501e-02, -1.32317841e-03]],\n",
       " \n",
       "         ...,\n",
       " \n",
       "         [[-3.42333838e-02, -2.54971441e-02,  2.32935511e-02, ...,\n",
       "           -2.04899926e-02,  1.97985061e-02, -3.63378040e-02],\n",
       "          [ 4.14710306e-02,  1.46116465e-02,  6.42932951e-04, ...,\n",
       "           -2.03389805e-02,  2.98547633e-02,  3.48540507e-02],\n",
       "          [-3.57929543e-02, -3.00128963e-02, -3.44088264e-02, ...,\n",
       "            5.39895147e-04,  3.43638770e-02,  4.14789356e-02]],\n",
       " \n",
       "         [[-1.05488300e-02, -1.57929678e-02,  8.88567045e-03, ...,\n",
       "           -3.91459279e-02, -3.45574543e-02, -1.74801089e-02],\n",
       "          [-1.01816431e-02, -1.49839185e-02,  7.62885436e-03, ...,\n",
       "           -2.55172439e-02, -1.92571599e-02, -2.13574450e-02],\n",
       "          [-3.56643759e-02,  3.81045081e-02,  3.18371169e-02, ...,\n",
       "            2.43514292e-02, -3.51217277e-02, -1.27563588e-02]],\n",
       " \n",
       "         [[ 2.16832012e-03, -2.26892252e-02, -1.20808631e-02, ...,\n",
       "            2.28189044e-02, -2.73818076e-02,  2.80601270e-02],\n",
       "          [ 2.78137140e-02,  2.01792754e-02, -3.54571417e-02, ...,\n",
       "           -2.23667920e-03,  1.85282230e-02,  3.64274047e-02],\n",
       "          [ 1.24740154e-02,  8.01938027e-03, -1.87698975e-02, ...,\n",
       "           -2.20799185e-02, -9.26223025e-03, -2.69784499e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 2.37193368e-02, -2.78566033e-02,  3.07529308e-02, ...,\n",
       "           -2.24381331e-02,  1.67898946e-02, -3.14069167e-03],\n",
       "          [-2.76896302e-02, -3.25629972e-02,  9.34448093e-04, ...,\n",
       "            2.78367102e-03, -3.08070444e-02,  1.25457346e-03],\n",
       "          [ 9.07018408e-03,  2.82911323e-02, -2.21735574e-02, ...,\n",
       "            2.84314789e-02, -1.01816356e-02,  1.72844157e-03]],\n",
       " \n",
       "         [[-2.42807530e-02,  1.50254518e-02, -3.65389958e-02, ...,\n",
       "            1.98144354e-02,  4.07866426e-02,  2.03118883e-02],\n",
       "          [-1.58730801e-02,  4.25615907e-03,  4.45052981e-03, ...,\n",
       "            3.02129425e-02,  1.45426393e-03, -3.99538241e-02],\n",
       "          [ 3.58556472e-02,  1.27275661e-02,  2.38328539e-02, ...,\n",
       "           -2.86678746e-02, -3.83174419e-03, -3.06816660e-02]],\n",
       " \n",
       "         [[ 3.44279371e-02, -2.67560035e-03, -3.70499231e-02, ...,\n",
       "            3.16773355e-03,  3.53251807e-02,  3.20744403e-02],\n",
       "          [ 2.09909938e-02,  1.51225142e-02,  9.75780934e-03, ...,\n",
       "           -3.25206965e-02, -1.60473716e-02, -2.92423144e-02],\n",
       "          [ 1.49294101e-02, -3.20325904e-02, -3.93646173e-02, ...,\n",
       "            2.07151212e-02,  3.15877423e-03,  2.29124017e-02]],\n",
       " \n",
       "         ...,\n",
       " \n",
       "         [[-4.12211902e-02,  1.35566294e-03, -2.98189856e-02, ...,\n",
       "           -1.28381345e-02, -2.26447042e-02,  3.34487744e-02],\n",
       "          [ 3.15556489e-02, -1.39483102e-02, -4.26370651e-04, ...,\n",
       "            2.56709196e-02, -3.70425843e-02,  2.46250518e-02],\n",
       "          [-3.60618345e-02, -3.12062427e-02,  4.19479720e-02, ...,\n",
       "            2.56344266e-02,  2.14696862e-02,  2.92077623e-02]],\n",
       " \n",
       "         [[ 4.14337404e-02, -3.61227952e-02, -2.99956091e-02, ...,\n",
       "            2.16290615e-02,  7.63043389e-03,  1.99841671e-02],\n",
       "          [ 7.26588070e-03,  2.94131823e-02,  1.09053329e-02, ...,\n",
       "           -1.10675022e-03, -2.36834921e-02,  2.41203941e-02],\n",
       "          [-1.94726493e-02,  2.50910781e-02, -3.61735225e-02, ...,\n",
       "            2.28871442e-02, -1.22174211e-02, -3.53127345e-02]],\n",
       " \n",
       "         [[ 5.57608530e-03,  3.72032411e-02, -3.08819897e-02, ...,\n",
       "            2.04905905e-02,  1.55304484e-02, -1.50547456e-02],\n",
       "          [-8.82233307e-03,  2.73670144e-02, -3.73219401e-02, ...,\n",
       "           -3.46564837e-02, -3.94804776e-02, -9.43819433e-03],\n",
       "          [ 1.34133063e-02, -1.50406398e-02,  2.09274255e-02, ...,\n",
       "           -4.13487405e-02,  1.10371225e-02,  1.16154552e-04]]],\n",
       " \n",
       " \n",
       "        [[[ 4.21794541e-02, -3.54188383e-02, -6.12611324e-03, ...,\n",
       "           -1.74785778e-03, -3.03587317e-03,  3.97468843e-02],\n",
       "          [-2.04522498e-02, -3.91669646e-02, -1.30606443e-03, ...,\n",
       "            3.26610357e-03, -4.01102081e-02,  7.79816136e-03],\n",
       "          [ 3.65524776e-02,  2.39172764e-02,  3.04996483e-02, ...,\n",
       "            1.89915858e-02,  3.44530605e-02, -4.87393513e-03]],\n",
       " \n",
       "         [[-3.04711405e-02, -3.28661092e-02, -1.66216772e-02, ...,\n",
       "            6.30413741e-03, -1.88671034e-02, -3.96364219e-02],\n",
       "          [ 1.33431703e-02, -2.28709169e-02,  1.72118843e-02, ...,\n",
       "           -3.96038666e-02,  3.75749357e-02,  8.47463682e-03],\n",
       "          [ 8.74791667e-03, -3.84564772e-02, -2.36325506e-02, ...,\n",
       "           -3.72837894e-02,  1.98512487e-02,  6.86955452e-03]],\n",
       " \n",
       "         [[ 1.15551129e-02, -1.83315948e-03, -3.57307494e-04, ...,\n",
       "           -3.84640507e-02, -1.96080264e-02, -1.73907317e-02],\n",
       "          [-1.22535937e-02, -1.11984462e-02,  1.81175284e-02, ...,\n",
       "            7.49651343e-03,  1.71403624e-02,  3.68377902e-02],\n",
       "          [ 1.73954293e-02, -4.04353999e-02, -4.68029082e-03, ...,\n",
       "           -3.23599502e-02, -3.29496078e-02,  4.12719585e-02]],\n",
       " \n",
       "         ...,\n",
       " \n",
       "         [[ 2.94711329e-02,  1.10547431e-02,  3.74296866e-02, ...,\n",
       "            2.66560577e-02, -3.12151518e-02, -2.98511609e-02],\n",
       "          [ 2.31104344e-03,  1.59617737e-02,  7.32618943e-03, ...,\n",
       "            3.10484730e-02,  2.17198618e-02,  2.74964832e-02],\n",
       "          [-2.24428251e-03,  1.72036886e-03, -3.83925512e-02, ...,\n",
       "           -1.96079537e-02, -4.10488658e-02, -2.72415578e-04]],\n",
       " \n",
       "         [[-4.17499356e-02,  3.42710726e-02,  2.57217176e-02, ...,\n",
       "           -3.00627481e-02,  2.98990868e-02,  7.23917410e-03],\n",
       "          [ 7.73192942e-03, -6.95761666e-03,  4.01291959e-02, ...,\n",
       "           -5.84744290e-03,  1.73882432e-02,  4.08782922e-02],\n",
       "          [ 8.52183998e-03, -2.59844176e-02,  1.53854005e-02, ...,\n",
       "           -4.00384851e-02,  1.04147680e-02,  3.27038281e-02]],\n",
       " \n",
       "         [[ 2.64701992e-03, -2.40799394e-02,  3.08555029e-02, ...,\n",
       "            3.79580818e-02,  1.58804171e-02,  1.10810399e-02],\n",
       "          [ 2.17963420e-02,  4.18467261e-02,  3.85223739e-02, ...,\n",
       "            1.71004608e-03,  1.98700167e-02, -1.42005533e-02],\n",
       "          [ 4.23698761e-02,  1.55120939e-02, -3.57375294e-02, ...,\n",
       "            4.12868373e-02,  3.23689543e-02,  4.63981554e-03]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.20673975,  0.00767195, -0.15854472, ..., -0.18725415,\n",
       "           -0.21016169,  0.05087964],\n",
       "          [ 0.02604149,  0.09467129, -0.08114992, ...,  0.09654279,\n",
       "           -0.11607575,  0.15857796],\n",
       "          [-0.09827623,  0.02595998,  0.15231733, ...,  0.0573913 ,\n",
       "           -0.19198617, -0.09082949],\n",
       "          ...,\n",
       "          [-0.17760535, -0.0816509 ,  0.07398112, ..., -0.02507751,\n",
       "           -0.08286265,  0.13323177],\n",
       "          [-0.03407699, -0.18438154,  0.19267277, ..., -0.07044515,\n",
       "            0.21055914,  0.0122501 ],\n",
       "          [-0.05354561,  0.18710022, -0.07546382, ...,  0.16850556,\n",
       "            0.17830251, -0.10003489]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 1.25923008e-03, -4.50532325e-02, -3.26332562e-02, ...,\n",
       "            5.48157096e-03, -1.41934715e-02,  1.19714215e-02],\n",
       "          [-5.31991981e-02, -4.32984531e-03,  9.58558917e-03, ...,\n",
       "            2.67924443e-02, -6.26078993e-02,  7.37188011e-03],\n",
       "          [-4.90036085e-02,  3.58109102e-02, -3.83730978e-02, ...,\n",
       "           -3.20215523e-02, -7.07946494e-02, -1.64884776e-02],\n",
       "          ...,\n",
       "          [ 4.06467691e-02,  4.35775146e-02,  6.66340292e-02, ...,\n",
       "           -4.78620715e-02, -2.30958126e-02, -2.94609852e-02],\n",
       "          [-5.08806370e-02,  2.98634768e-02,  5.59884906e-02, ...,\n",
       "           -5.32659590e-02, -1.52585320e-02,  3.91936898e-02],\n",
       "          [ 3.42974588e-02, -3.16472463e-02,  3.95853892e-02, ...,\n",
       "            3.33802402e-02, -5.78014031e-02, -1.08303316e-02]],\n",
       " \n",
       "         [[ 6.60726875e-02, -4.17507105e-02, -1.98468342e-02, ...,\n",
       "            2.42851377e-02,  4.04313952e-03,  6.16179407e-03],\n",
       "          [-3.49012651e-02, -1.21155083e-02,  5.33361733e-03, ...,\n",
       "           -2.22282484e-02, -6.28052577e-02, -7.08497390e-02],\n",
       "          [ 2.29239687e-02, -3.33002284e-02, -6.37158602e-02, ...,\n",
       "           -5.69671877e-02, -2.37862878e-02, -2.54591778e-02],\n",
       "          ...,\n",
       "          [ 1.84352919e-02, -5.00287861e-03, -1.08915009e-02, ...,\n",
       "           -6.16139770e-02, -6.69426173e-02, -5.25878370e-02],\n",
       "          [ 4.63570431e-02, -5.86393215e-02, -5.94411567e-02, ...,\n",
       "            6.27798289e-02,  7.19399601e-02,  3.80924121e-02],\n",
       "          [-1.53276138e-02,  4.19576317e-02,  4.09426466e-02, ...,\n",
       "           -1.66745484e-03,  5.63049316e-03,  3.67177203e-02]],\n",
       " \n",
       "         [[ 4.52129766e-02,  2.76900530e-02, -1.86571516e-02, ...,\n",
       "            1.09235197e-02,  3.04257795e-02, -5.56626543e-02],\n",
       "          [-1.14283226e-02, -6.12281561e-02, -2.47064307e-02, ...,\n",
       "           -1.32198632e-03, -6.21356070e-02, -1.98366493e-02],\n",
       "          [ 4.81697917e-02,  2.86669806e-02, -6.28776103e-03, ...,\n",
       "            1.97529569e-02, -1.67147592e-02, -3.80918421e-02],\n",
       "          ...,\n",
       "          [-4.03282568e-02, -5.23270927e-02,  2.31090933e-02, ...,\n",
       "           -2.17995346e-02, -1.66972429e-02,  3.03923860e-02],\n",
       "          [-1.05120987e-03, -2.88676061e-02,  2.80313417e-02, ...,\n",
       "            5.61105460e-02,  1.61249042e-02, -1.75242163e-02],\n",
       "          [ 2.18963400e-02,  4.10763919e-02,  5.63131869e-02, ...,\n",
       "            5.58607876e-02, -1.67241544e-02, -1.27985850e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 1.91297233e-02,  6.24183118e-02, -3.78361717e-02, ...,\n",
       "           -2.62754634e-02, -1.66043639e-03, -2.58115456e-02],\n",
       "          [ 3.03350687e-02,  4.82804105e-02, -6.46192580e-02, ...,\n",
       "            5.10231033e-02,  7.45722651e-03,  1.11351088e-02],\n",
       "          [ 1.32916197e-02,  1.60905123e-02,  4.84658256e-02, ...,\n",
       "            1.04574189e-02,  1.98194459e-02, -1.28448196e-02],\n",
       "          ...,\n",
       "          [ 5.35162836e-02, -1.46347992e-02,  6.02992475e-02, ...,\n",
       "           -6.97693527e-02,  4.39367294e-02, -5.87408729e-02],\n",
       "          [ 6.71345443e-02, -2.67489143e-02,  8.74683261e-04, ...,\n",
       "           -2.16587521e-02, -2.47855820e-02, -5.44142611e-02],\n",
       "          [-6.01293594e-02,  2.19855011e-02, -2.72406042e-02, ...,\n",
       "           -3.32760848e-02, -1.00913644e-03, -7.20351040e-02]],\n",
       " \n",
       "         [[-3.19933333e-02,  6.23575747e-02,  4.94763926e-02, ...,\n",
       "            6.66331798e-02, -2.13477984e-02, -3.78504880e-02],\n",
       "          [ 4.09912914e-02,  4.46878225e-02,  6.00478798e-03, ...,\n",
       "           -2.67284214e-02,  3.75966281e-02, -3.15036401e-02],\n",
       "          [ 6.34018779e-02, -4.18432429e-02, -2.60769725e-02, ...,\n",
       "            5.35185933e-02, -4.57128398e-02, -5.35156913e-02],\n",
       "          ...,\n",
       "          [ 5.06839007e-02, -4.64296751e-02, -6.50263578e-03, ...,\n",
       "            4.15942073e-03, -6.86065629e-02,  1.85076445e-02],\n",
       "          [-1.23743117e-02, -1.19470097e-02, -1.78187862e-02, ...,\n",
       "           -3.93129289e-02,  3.26280221e-02, -1.66270062e-02],\n",
       "          [ 7.03331828e-02, -6.22850060e-02, -2.00724602e-03, ...,\n",
       "            1.71499401e-02, -4.37896661e-02,  7.75456429e-05]],\n",
       " \n",
       "         [[-1.53285600e-02,  2.84646526e-02, -9.04656947e-04, ...,\n",
       "            2.45226026e-02, -5.88875562e-02,  6.93992674e-02],\n",
       "          [ 4.74777520e-02,  5.11939153e-02,  6.13298863e-02, ...,\n",
       "           -5.03211543e-02, -4.92166281e-02,  5.62186986e-02],\n",
       "          [ 1.80188268e-02, -4.93273139e-02, -3.98978926e-02, ...,\n",
       "            9.68204439e-03, -6.85696304e-03,  8.97035748e-03],\n",
       "          ...,\n",
       "          [-1.75399259e-02, -5.85329346e-02,  3.30686271e-02, ...,\n",
       "           -5.10812476e-02,  3.81271467e-02,  4.04642224e-02],\n",
       "          [-3.73025499e-02, -1.66729651e-02,  3.60476375e-02, ...,\n",
       "            3.91910747e-02,  2.81426460e-02, -3.90602164e-02],\n",
       "          [ 6.58143610e-02, -3.81134711e-02, -7.34621286e-03, ...,\n",
       "            5.03859520e-02,  4.53508198e-02,  1.59837604e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 1.58463493e-02,  6.34236783e-02,  3.42908502e-02, ...,\n",
       "           -3.65922377e-02,  4.18968797e-02, -1.57222934e-02],\n",
       "          [-2.22294368e-02,  3.91472504e-02, -1.81843713e-02, ...,\n",
       "            1.48310065e-02, -1.59975626e-02, -3.32079828e-04],\n",
       "          [-5.63025922e-02,  3.19871083e-02,  1.74615681e-02, ...,\n",
       "           -5.95744029e-02,  1.27900019e-02,  4.31015119e-02],\n",
       "          ...,\n",
       "          [-2.41306722e-02,  1.09411553e-02, -1.98544413e-02, ...,\n",
       "            1.39069185e-02, -3.30546498e-03, -5.65126836e-02],\n",
       "          [-1.47506595e-04, -2.97550410e-02,  7.09980726e-02, ...,\n",
       "           -3.54147367e-02,  4.36070040e-02, -4.85791266e-03],\n",
       "          [-3.60366441e-02, -1.84840038e-02,  3.18256766e-02, ...,\n",
       "            4.92971539e-02,  5.55001199e-02, -3.97489704e-02]],\n",
       " \n",
       "         [[-1.78196281e-03,  3.18569392e-02, -3.38853635e-02, ...,\n",
       "           -5.38801923e-02, -5.73695600e-02, -6.27873465e-02],\n",
       "          [-6.10603616e-02,  6.66769594e-02, -5.67913055e-04, ...,\n",
       "            4.38628942e-02, -3.18513811e-02,  2.22973153e-02],\n",
       "          [ 1.78393498e-02, -3.37918103e-03,  6.07627928e-02, ...,\n",
       "            4.34942171e-02, -2.91533023e-03, -2.40987539e-03],\n",
       "          ...,\n",
       "          [-5.58641031e-02,  2.59654075e-02, -4.96419519e-02, ...,\n",
       "            7.97899067e-03, -5.10139540e-02, -4.97215465e-02],\n",
       "          [ 5.21661416e-02,  8.27927887e-03, -5.61247617e-02, ...,\n",
       "           -2.56941468e-03,  2.19387487e-02, -1.51012130e-02],\n",
       "          [-4.59198505e-02, -3.11954059e-02,  6.53433949e-02, ...,\n",
       "            6.64288998e-02, -5.76628596e-02,  3.88637185e-02]],\n",
       " \n",
       "         [[ 1.08815581e-02,  3.43272910e-02, -6.17636554e-02, ...,\n",
       "            5.25247082e-02,  4.14283648e-02, -4.40225005e-02],\n",
       "          [ 1.78822130e-03, -6.11345731e-02,  5.03423735e-02, ...,\n",
       "            4.35488299e-02,  5.29999584e-02,  4.22812328e-02],\n",
       "          [ 4.64906394e-02,  3.78462747e-02, -2.60845274e-02, ...,\n",
       "            4.48891521e-02, -7.17083588e-02, -6.61023259e-02],\n",
       "          ...,\n",
       "          [ 1.76793486e-02,  5.01279309e-02,  2.43504345e-02, ...,\n",
       "           -3.60258706e-02, -1.15144551e-02, -1.54030807e-02],\n",
       "          [-1.38204396e-02, -6.95277601e-02, -5.11421226e-02, ...,\n",
       "            5.79330176e-02,  2.92528942e-02, -1.14605650e-02],\n",
       "          [ 1.17359236e-02,  3.48694474e-02,  5.18374890e-03, ...,\n",
       "            2.83066928e-02,  1.79691017e-02, -2.76230685e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.12126303, -0.12648876,  0.087291  , ..., -0.02062847,\n",
       "            0.10363798, -0.09848762],\n",
       "          [ 0.0166624 ,  0.06263846,  0.13295141, ...,  0.07651466,\n",
       "            0.01636101, -0.13321778],\n",
       "          [ 0.0934103 , -0.0741208 ,  0.11010855, ..., -0.02098517,\n",
       "           -0.05516759,  0.07060708],\n",
       "          ...,\n",
       "          [ 0.03527726, -0.13407947, -0.03354655, ...,  0.04214017,\n",
       "           -0.04976183,  0.02400076],\n",
       "          [ 0.05480994,  0.08543125,  0.04640219, ..., -0.11944338,\n",
       "           -0.03422198,  0.10904194],\n",
       "          [ 0.09212857, -0.06029856,  0.07347207, ..., -0.11847067,\n",
       "           -0.03826007, -0.07834718]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([[[[-0.07686417,  0.13652283, -0.01103762, ..., -0.12840098,\n",
       "           -0.06964903, -0.09263314],\n",
       "          [ 0.04161467,  0.12343496, -0.01583719, ...,  0.0395617 ,\n",
       "           -0.11555322, -0.02567146],\n",
       "          [-0.04611973,  0.07509965, -0.09700823, ...,  0.07965747,\n",
       "           -0.01614068, -0.00534393],\n",
       "          ...,\n",
       "          [-0.00464845,  0.07004327,  0.06808238, ..., -0.06875888,\n",
       "            0.09567383,  0.02054362],\n",
       "          [-0.04648177, -0.01520675, -0.02057529, ..., -0.00771101,\n",
       "            0.08544388, -0.0919043 ],\n",
       "          [-0.02417365,  0.00760302, -0.1226635 , ..., -0.12674706,\n",
       "            0.03351584,  0.07020837]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-4.74591553e-02, -8.98311287e-02,  1.20923728e-01, ...,\n",
       "            2.90300697e-02,  8.53586793e-02,  1.20052069e-01],\n",
       "          [-4.79692668e-02,  1.19870633e-01,  7.32728243e-02, ...,\n",
       "            2.88482159e-02,  8.59119147e-02, -2.08493918e-02],\n",
       "          [-6.24857321e-02,  1.30357057e-01,  9.90994275e-02, ...,\n",
       "           -9.99099463e-02, -4.18103784e-02, -4.58151326e-02],\n",
       "          ...,\n",
       "          [ 2.49920487e-02, -3.92400920e-02,  7.96914101e-05, ...,\n",
       "           -1.15905151e-01,  1.06839746e-01,  1.31603986e-01],\n",
       "          [ 1.04979634e-01, -8.59988779e-02,  1.03017733e-01, ...,\n",
       "           -1.20944485e-01, -1.00510255e-01, -8.83800685e-02],\n",
       "          [ 3.56996059e-03,  4.09357399e-02,  1.94859654e-02, ...,\n",
       "           -1.00493997e-01,  1.06433630e-01,  1.12284914e-01]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-6.33785948e-02,  4.79049981e-03,  1.43461302e-02, ...,\n",
       "           -1.43289566e-02,  6.64220452e-02, -2.59836093e-02],\n",
       "          [-1.43583864e-03, -4.04725336e-02,  8.98484141e-03, ...,\n",
       "           -6.56215847e-02,  3.50863859e-02,  3.90117615e-02],\n",
       "          [-4.52283621e-02,  8.92603397e-03,  4.85438555e-02, ...,\n",
       "           -4.04075794e-02, -6.04105629e-02, -4.12092432e-02],\n",
       "          ...,\n",
       "          [-1.51351951e-02,  2.39176080e-02,  4.32993174e-02, ...,\n",
       "           -5.31998873e-02,  6.83727413e-02, -1.76495463e-02],\n",
       "          [-1.91464648e-02,  6.02781326e-02, -7.50626624e-03, ...,\n",
       "           -3.27468701e-02, -7.18472451e-02,  3.37163284e-02],\n",
       "          [ 4.05264571e-02, -4.13070433e-02,  1.82556361e-03, ...,\n",
       "           -4.01345491e-02,  5.74278086e-03, -1.41937286e-03]],\n",
       " \n",
       "         [[ 6.35990649e-02,  1.22375861e-02, -4.14357483e-02, ...,\n",
       "            1.36875510e-02, -3.72369587e-02, -1.41609199e-02],\n",
       "          [ 3.00920978e-02, -3.01821567e-02,  4.24809828e-02, ...,\n",
       "           -3.48922834e-02,  1.34871528e-02, -3.95266488e-02],\n",
       "          [ 1.55935064e-02, -3.86837237e-02,  2.22156569e-02, ...,\n",
       "           -5.54267354e-02,  8.02813470e-03,  3.81411240e-02],\n",
       "          ...,\n",
       "          [ 6.12930954e-03, -7.13045225e-02,  3.78198251e-02, ...,\n",
       "           -7.50588626e-03,  3.83607745e-02,  3.20885032e-02],\n",
       "          [ 2.16062739e-02,  5.09200767e-02, -4.23506424e-02, ...,\n",
       "            8.61831009e-03, -8.78621638e-03,  4.08133939e-02],\n",
       "          [-1.66576169e-02,  1.06230602e-02,  5.17125353e-02, ...,\n",
       "            4.69779596e-02, -4.88657355e-02, -3.38579044e-02]],\n",
       " \n",
       "         [[-4.00588661e-03,  4.08651009e-02,  7.84829259e-03, ...,\n",
       "           -4.80586030e-02,  4.99404818e-02, -2.73974240e-02],\n",
       "          [ 6.46874607e-02, -3.10652219e-02,  2.51202807e-02, ...,\n",
       "            4.92682979e-02, -2.03510374e-03,  1.03728995e-02],\n",
       "          [ 2.26420611e-02, -3.50689404e-02,  6.18280470e-02, ...,\n",
       "            1.07163712e-02,  4.00362164e-02, -2.45464630e-02],\n",
       "          ...,\n",
       "          [ 6.72974735e-02, -1.89062208e-03,  4.91122901e-02, ...,\n",
       "            4.27508131e-02,  3.68889123e-02, -6.38099611e-02],\n",
       "          [-1.11488886e-02,  2.97825038e-02, -2.84739397e-02, ...,\n",
       "            4.07848656e-02,  5.74072748e-02,  4.08310294e-02],\n",
       "          [-6.24272525e-02, -5.15615642e-02, -4.70097512e-02, ...,\n",
       "            2.73147970e-02,  6.47989959e-02,  2.90528163e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 2.29497254e-03,  6.97242767e-02, -1.83698386e-02, ...,\n",
       "            6.54839128e-02, -4.25215550e-02, -5.35358414e-02],\n",
       "          [ 4.63413745e-02,  5.91476262e-02, -4.00750488e-02, ...,\n",
       "           -3.55321169e-02, -9.06912237e-03, -4.96545807e-02],\n",
       "          [ 1.13916919e-02, -4.93057743e-02, -1.94250904e-02, ...,\n",
       "            3.21517512e-02, -1.02713481e-02,  6.61526918e-02],\n",
       "          ...,\n",
       "          [ 4.07874584e-04,  6.54212385e-02, -6.80774152e-02, ...,\n",
       "           -3.42531353e-02, -4.46743295e-02,  5.14662415e-02],\n",
       "          [ 5.84901869e-02,  5.68634719e-02, -2.95290016e-02, ...,\n",
       "           -2.51476057e-02, -3.06018889e-02,  3.90539914e-02],\n",
       "          [ 3.87278423e-02,  4.62822169e-02,  5.61004728e-02, ...,\n",
       "           -5.76905273e-02, -2.57029235e-02, -5.48345074e-02]],\n",
       " \n",
       "         [[-6.30113259e-02, -5.35687208e-02,  6.73578382e-02, ...,\n",
       "            1.71271637e-02, -6.16233051e-03, -2.40557566e-02],\n",
       "          [ 3.37249115e-02,  5.51735908e-02, -3.52115445e-02, ...,\n",
       "            5.12135625e-02,  5.39215356e-02,  1.16359517e-02],\n",
       "          [-1.31211020e-02,  6.77306205e-02,  4.36484218e-02, ...,\n",
       "            3.56322080e-02,  1.53341517e-02,  5.79886585e-02],\n",
       "          ...,\n",
       "          [-2.27145106e-03,  3.41782868e-02,  7.11324811e-02, ...,\n",
       "            5.25725037e-02,  5.64797372e-02, -1.36236325e-02],\n",
       "          [-4.85529751e-02,  3.80141065e-02,  8.81164521e-03, ...,\n",
       "           -2.04777233e-02, -1.63718015e-02,  5.17323539e-02],\n",
       "          [-3.86584476e-02,  6.26367778e-02, -4.77060601e-02, ...,\n",
       "           -2.82682478e-04, -6.61993399e-02,  1.40026733e-02]],\n",
       " \n",
       "         [[ 5.48499227e-02, -5.14658764e-02, -3.14827338e-02, ...,\n",
       "           -2.65975185e-02, -6.11135811e-02, -3.64101604e-02],\n",
       "          [ 4.66672257e-02, -4.26723696e-02,  5.42961359e-02, ...,\n",
       "           -5.46845905e-02,  2.95081958e-02,  4.70384881e-02],\n",
       "          [ 1.99466869e-02,  7.98031688e-05,  6.58017844e-02, ...,\n",
       "            2.28624046e-02, -6.01776578e-02, -6.85659721e-02],\n",
       "          ...,\n",
       "          [-4.40662727e-02,  2.30234265e-02, -2.29552537e-02, ...,\n",
       "           -6.05486780e-02, -1.40993372e-02,  3.62200812e-02],\n",
       "          [ 2.85290033e-02,  5.80683649e-02,  2.33449936e-02, ...,\n",
       "            4.33543772e-02,  2.26600915e-02,  3.85561064e-02],\n",
       "          [-1.85795352e-02, -4.77998853e-02, -3.52656841e-03, ...,\n",
       "           -1.50106549e-02,  1.69028044e-02, -5.50402775e-02]]],\n",
       " \n",
       " \n",
       "        [[[-6.05039746e-02, -3.12099792e-02,  5.36372513e-03, ...,\n",
       "            1.74435377e-02,  3.64459977e-02,  4.06824499e-02],\n",
       "          [ 3.38832289e-02,  3.00829932e-02, -5.75005189e-02, ...,\n",
       "            1.69284940e-02,  7.16928542e-02,  3.61096337e-02],\n",
       "          [ 1.81563422e-02, -2.04306133e-02, -4.51656766e-02, ...,\n",
       "            3.81329283e-02, -2.93074548e-03,  2.14307010e-02],\n",
       "          ...,\n",
       "          [ 5.19190729e-03,  3.00752372e-03,  2.77726948e-02, ...,\n",
       "           -4.22801822e-03,  6.06107265e-02,  2.04636008e-02],\n",
       "          [-1.67931169e-02,  6.63772076e-02, -5.84316067e-02, ...,\n",
       "           -6.26754239e-02, -1.81970373e-02, -4.55072522e-03],\n",
       "          [-1.24673992e-02, -2.70774961e-03, -5.31651638e-02, ...,\n",
       "           -6.43498600e-02,  1.07735991e-02,  6.86464608e-02]],\n",
       " \n",
       "         [[ 3.44322696e-02,  4.17107195e-02, -1.41388252e-02, ...,\n",
       "            5.27260527e-02,  2.29332149e-02, -1.18440464e-02],\n",
       "          [ 6.04384691e-02, -2.57689767e-02, -1.58772357e-02, ...,\n",
       "           -4.62214798e-02, -5.34066893e-02,  6.47041053e-02],\n",
       "          [-7.56805390e-03,  6.44413978e-02,  6.78518414e-03, ...,\n",
       "           -2.67524272e-02, -3.98686081e-02, -1.62312426e-02],\n",
       "          ...,\n",
       "          [ 1.22096613e-02,  2.99392343e-02,  1.60663202e-02, ...,\n",
       "           -4.32571769e-03,  1.78313628e-02,  9.86102968e-03],\n",
       "          [ 2.10758522e-02, -6.52299151e-02, -3.03490572e-02, ...,\n",
       "            6.96274340e-02, -7.65458494e-03,  4.99773696e-02],\n",
       "          [ 5.23628145e-02, -5.96717224e-02,  2.54664347e-02, ...,\n",
       "           -1.37735493e-02, -7.03027323e-02,  3.07283551e-03]],\n",
       " \n",
       "         [[ 2.30897218e-03,  2.27565020e-02, -2.58850195e-02, ...,\n",
       "           -4.51061092e-02, -3.20006981e-02,  4.24610078e-02],\n",
       "          [-7.08690137e-02, -2.20859349e-02,  2.60398388e-02, ...,\n",
       "            5.75352460e-02, -2.52551436e-02,  1.92756206e-02],\n",
       "          [-4.03703265e-02, -6.35195151e-02,  4.74310666e-02, ...,\n",
       "           -6.49509281e-02, -3.06589454e-02, -1.67381912e-02],\n",
       "          ...,\n",
       "          [ 6.85410947e-02,  4.93809804e-02, -5.70218340e-02, ...,\n",
       "            6.15215600e-02, -1.05714612e-02, -3.23660746e-02],\n",
       "          [ 2.98980623e-02, -6.01013638e-02, -5.89362681e-02, ...,\n",
       "            4.56705317e-02,  1.39718726e-02, -5.33957295e-02],\n",
       "          [-5.39005101e-02, -1.17918402e-02,  7.07416981e-02, ...,\n",
       "           -2.84441747e-02,  5.34235686e-03, -3.13959643e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.09768696,  0.05854803, -0.09965585, ..., -0.0902936 ,\n",
       "           -0.05991258, -0.0393806 ],\n",
       "          [ 0.06048632, -0.04004408,  0.08291584, ...,  0.05542514,\n",
       "           -0.05672076,  0.07230188],\n",
       "          [-0.07859792, -0.07222944, -0.09338698, ...,  0.04339632,\n",
       "            0.02547823, -0.03596428],\n",
       "          ...,\n",
       "          [-0.02464828,  0.04453821, -0.09610927, ...,  0.03457895,\n",
       "            0.12241232,  0.10410036],\n",
       "          [ 0.05655576,  0.11513132,  0.09979343, ...,  0.1120144 ,\n",
       "           -0.01859053,  0.09330325],\n",
       "          [-0.12315183, -0.12763968, -0.08168039, ...,  0.00906552,\n",
       "            0.06546208,  0.03744684]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 0.07970294, -0.05138899,  0.12785715, ..., -0.03973244,\n",
       "            0.0517507 ,  0.07600063],\n",
       "          [ 0.11906013, -0.09466083,  0.05191642, ...,  0.09631759,\n",
       "           -0.05929474,  0.05040801],\n",
       "          [ 0.01325709, -0.04634688,  0.0385422 , ...,  0.12298259,\n",
       "            0.01194814,  0.06171794],\n",
       "          ...,\n",
       "          [-0.03633247,  0.07697673, -0.0801422 , ...,  0.01656085,\n",
       "            0.01807539,  0.02407572],\n",
       "          [ 0.10923211, -0.09827548,  0.10454093, ...,  0.09864902,\n",
       "           -0.12525854, -0.06059512],\n",
       "          [-0.01148199,  0.06310201, -0.12253177, ..., -0.08524523,\n",
       "            0.10451664,  0.00887176]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-9.59683955e-03, -6.26864657e-02, -8.79880786e-03, ...,\n",
       "           -5.24118505e-02, -6.69689998e-02, -2.94870883e-03],\n",
       "          [ 9.93311405e-05, -1.73625611e-02, -3.76217812e-02, ...,\n",
       "            4.24959511e-02,  1.31553933e-02, -6.29156083e-02],\n",
       "          [-5.12915254e-02, -4.63363305e-02,  1.64044723e-02, ...,\n",
       "           -4.64132763e-02,  6.63902611e-03, -4.94977087e-02],\n",
       "          ...,\n",
       "          [ 5.48119843e-03, -8.63032043e-03, -1.17539875e-02, ...,\n",
       "           -1.49502605e-03,  6.94958568e-02,  3.56253237e-03],\n",
       "          [-5.91373779e-02, -6.32664263e-02, -5.71077652e-02, ...,\n",
       "           -3.61107849e-02, -6.31441772e-02,  6.88793361e-02],\n",
       "          [-5.49071282e-03,  4.06699255e-02, -1.63256861e-02, ...,\n",
       "           -5.46550006e-03, -4.35317941e-02, -3.47687751e-02]],\n",
       " \n",
       "         [[-4.32281345e-02,  2.30211541e-02, -2.66562253e-02, ...,\n",
       "           -8.55583698e-03, -5.57831489e-02, -2.28288732e-02],\n",
       "          [ 3.95116806e-02,  5.72989881e-03, -6.01223744e-02, ...,\n",
       "            6.13836199e-02, -5.13105243e-02, -6.48967922e-03],\n",
       "          [-6.65851086e-02, -6.90628439e-02, -1.21251643e-02, ...,\n",
       "           -6.38810545e-03, -3.68722528e-02,  3.82788256e-02],\n",
       "          ...,\n",
       "          [ 3.65825817e-02,  4.42552194e-02, -6.88613355e-02, ...,\n",
       "           -2.56434754e-02, -4.01235707e-02, -2.81542651e-02],\n",
       "          [ 2.26486325e-02,  3.52269113e-02, -3.93059589e-02, ...,\n",
       "            2.49218941e-03,  1.46665424e-02, -6.30437732e-02],\n",
       "          [-5.10348603e-02,  2.76301056e-02,  2.48651952e-02, ...,\n",
       "           -5.36993369e-02,  6.00425601e-02,  6.22649640e-02]],\n",
       " \n",
       "         [[ 3.78233343e-02,  6.90511763e-02, -1.36749074e-02, ...,\n",
       "            3.53855565e-02,  1.33461133e-02, -6.91025853e-02],\n",
       "          [ 4.73603532e-02, -3.89301032e-02, -1.53962858e-02, ...,\n",
       "            6.41644597e-02, -2.25395821e-02, -1.75833702e-04],\n",
       "          [ 7.11872131e-02,  5.99857271e-02,  2.80039832e-02, ...,\n",
       "            2.81931534e-02, -5.17239422e-02,  4.51758802e-02],\n",
       "          ...,\n",
       "          [ 2.01269388e-02,  1.45555437e-02, -1.99599825e-02, ...,\n",
       "            1.40552223e-03,  1.40842795e-03,  4.11642641e-02],\n",
       "          [ 7.16189593e-02,  5.38905710e-03, -1.41593367e-02, ...,\n",
       "           -2.44434141e-02, -1.97004601e-02, -6.68062046e-02],\n",
       "          [ 1.76073015e-02,  4.68294472e-02,  1.67607516e-03, ...,\n",
       "           -2.28780136e-02, -3.59793119e-02,  4.94654328e-02]]],\n",
       " \n",
       " \n",
       "        [[[-4.59808484e-02,  4.15278375e-02, -6.34477958e-02, ...,\n",
       "           -6.26086444e-02, -2.53925212e-02, -7.62376934e-03],\n",
       "          [ 1.64826959e-03, -4.03828174e-03,  1.86748579e-02, ...,\n",
       "           -6.90119639e-02, -4.19966429e-02,  5.46103567e-02],\n",
       "          [-3.77014279e-03, -6.82083219e-02, -4.12236787e-02, ...,\n",
       "            3.51927057e-02, -7.96830654e-03,  5.85515350e-02],\n",
       "          ...,\n",
       "          [-2.71043479e-02, -3.06693204e-02, -3.92263792e-02, ...,\n",
       "           -1.08382478e-02,  5.21815568e-02, -5.73577024e-02],\n",
       "          [-6.45436198e-02, -2.78844684e-03,  3.81098092e-02, ...,\n",
       "           -6.98007569e-02, -3.68708372e-03,  1.96145475e-03],\n",
       "          [ 2.51764953e-02, -5.25179617e-02, -3.40533257e-03, ...,\n",
       "           -3.49021070e-02, -7.00555146e-02, -6.29662126e-02]],\n",
       " \n",
       "         [[ 6.14800453e-02, -9.60988551e-03,  3.13574076e-03, ...,\n",
       "            9.42758471e-03,  1.13651380e-02, -1.08510666e-02],\n",
       "          [ 1.72345638e-02, -4.88313586e-02,  4.85405698e-02, ...,\n",
       "            3.17247435e-02, -4.49792445e-02,  9.02844220e-03],\n",
       "          [-2.99270526e-02, -6.02490976e-02, -3.51605304e-02, ...,\n",
       "           -6.27970845e-02, -5.42742535e-02,  3.22408453e-02],\n",
       "          ...,\n",
       "          [-6.76919892e-02, -1.36430934e-02,  4.84572053e-02, ...,\n",
       "           -4.41849828e-02, -2.96304859e-02,  6.30498081e-02],\n",
       "          [-3.25514749e-02, -6.53785467e-02,  6.22785389e-02, ...,\n",
       "           -6.81446418e-02, -3.69158909e-02,  1.34080723e-02],\n",
       "          [ 6.32142425e-02, -1.63509808e-02, -6.31707236e-02, ...,\n",
       "           -1.63689964e-02,  1.96883306e-02,  2.44077146e-02]],\n",
       " \n",
       "         [[ 3.16194594e-02, -5.57628423e-02,  5.01421392e-02, ...,\n",
       "           -1.09712519e-02, -2.03781836e-02,  4.07480747e-02],\n",
       "          [-3.39526683e-03, -2.14912295e-02,  3.61426920e-03, ...,\n",
       "            6.99713081e-02,  5.97647876e-02, -6.32927194e-02],\n",
       "          [ 6.51399791e-02,  6.65677339e-02, -4.35972959e-02, ...,\n",
       "            6.23078048e-02, -4.31362540e-02, -4.48957235e-02],\n",
       "          ...,\n",
       "          [-7.01407045e-02,  2.59285867e-02, -1.38211437e-02, ...,\n",
       "            3.21592242e-02, -3.86197492e-02,  8.85526091e-03],\n",
       "          [ 7.04302043e-02, -4.24991548e-02,  1.94282234e-02, ...,\n",
       "           -1.36089362e-02, -2.79791728e-02,  1.57466382e-02],\n",
       "          [ 3.03232670e-02,  1.19166374e-02, -8.74736160e-03, ...,\n",
       "           -2.27890573e-02, -4.36728857e-02, -6.14605471e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 3.14489603e-02, -8.21862370e-03,  4.81387153e-02, ...,\n",
       "            2.18449458e-02, -1.81765780e-02,  9.48002934e-03],\n",
       "          [ 5.56898564e-02, -3.00057568e-02,  5.97724766e-02, ...,\n",
       "            4.71567437e-02, -2.50839926e-02, -2.12994479e-02],\n",
       "          [-7.20075071e-02, -2.30148733e-02, -4.66899723e-02, ...,\n",
       "           -2.55838037e-02, -4.49737050e-02, -4.23349366e-02],\n",
       "          ...,\n",
       "          [-2.54129618e-03, -6.11894801e-02, -1.22117437e-02, ...,\n",
       "           -7.71433860e-03, -2.46517360e-04, -3.65828611e-02],\n",
       "          [ 5.65763414e-02,  3.06010470e-02, -4.72324379e-02, ...,\n",
       "           -1.33435801e-02, -1.58403292e-02, -3.11409123e-02],\n",
       "          [ 2.25022957e-02, -4.76388857e-02, -8.30668956e-03, ...,\n",
       "            3.94735113e-02, -6.91123754e-02,  3.56515497e-02]],\n",
       " \n",
       "         [[-3.71223651e-02, -2.33798549e-02, -5.00248596e-02, ...,\n",
       "            6.19339794e-02, -2.75849923e-02,  1.32365525e-03],\n",
       "          [ 2.40683705e-02, -1.47870183e-03,  1.66604221e-02, ...,\n",
       "            4.01099473e-02,  7.19583631e-02,  4.69665304e-02],\n",
       "          [-1.57256499e-02, -3.75177674e-02, -2.10687630e-02, ...,\n",
       "           -4.30670381e-03, -3.02883536e-02,  3.83333340e-02],\n",
       "          ...,\n",
       "          [ 1.18609965e-02, -5.56217134e-03,  5.97519577e-02, ...,\n",
       "            5.19477278e-02, -3.33119109e-02, -1.34169310e-02],\n",
       "          [-5.94205409e-02,  6.40665442e-02,  3.44680250e-03, ...,\n",
       "            1.31307021e-02, -1.53220892e-02, -2.85801217e-02],\n",
       "          [ 2.91993991e-02, -2.96696126e-02,  4.42875475e-02, ...,\n",
       "           -5.61483353e-02,  2.20543966e-02, -3.85642797e-03]],\n",
       " \n",
       "         [[-3.82515006e-02, -4.71018255e-02, -3.85193005e-02, ...,\n",
       "            1.01385489e-02,  3.87512743e-02, -4.38109338e-02],\n",
       "          [ 6.94406629e-02, -5.29829115e-02,  2.82841176e-02, ...,\n",
       "           -2.80091614e-02,  6.40648752e-02, -1.60504878e-03],\n",
       "          [ 2.19555944e-02, -6.82450756e-02, -1.41521767e-02, ...,\n",
       "           -5.18348515e-02, -4.85319644e-03,  7.06922561e-02],\n",
       "          ...,\n",
       "          [-4.56277877e-02,  2.49072686e-02,  9.24029201e-03, ...,\n",
       "           -2.31040269e-03, -1.66878998e-02, -5.34724183e-02],\n",
       "          [ 1.50974467e-02, -1.87518224e-02,  1.53853074e-02, ...,\n",
       "            1.59599632e-02, -2.92368904e-02,  2.79065296e-02],\n",
       "          [ 3.20654958e-02,  2.57722437e-02,  4.31426913e-02, ...,\n",
       "           -6.08723611e-03, -4.32366878e-03, -2.83741951e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.06723228,  0.03858605,  0.09041286, ..., -0.02854507,\n",
       "           -0.0072647 ,  0.03872783],\n",
       "          [ 0.01731534, -0.08353734, -0.0183812 , ..., -0.13568096,\n",
       "           -0.04226841,  0.05562465],\n",
       "          [ 0.12054679,  0.09462698, -0.07583498, ..., -0.0501995 ,\n",
       "            0.06060889,  0.12171268],\n",
       "          ...,\n",
       "          [ 0.12464863, -0.00973678, -0.02952953, ..., -0.09907151,\n",
       "           -0.08263133, -0.01649059],\n",
       "          [ 0.0451887 ,  0.1297518 , -0.09722181, ...,  0.05229105,\n",
       "           -0.08642495, -0.06862552],\n",
       "          [ 0.05869573,  0.08316745, -0.00196762, ...,  0.06281811,\n",
       "           -0.07108311,  0.04461941]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-0.09527114, -0.09566817,  0.05926126, ..., -0.11750385,\n",
       "           -0.07516667,  0.05406395],\n",
       "          [-0.00430864,  0.1021229 ,  0.05400568, ..., -0.01213175,\n",
       "           -0.08069831, -0.09193623],\n",
       "          [ 0.11789089, -0.01049596,  0.06716385, ..., -0.02813345,\n",
       "            0.08086273, -0.10573545],\n",
       "          ...,\n",
       "          [ 0.07578617,  0.11320177, -0.07311347, ..., -0.11961228,\n",
       "            0.06299701, -0.02344871],\n",
       "          [ 0.11399001,  0.09164447, -0.06136718, ...,  0.02518606,\n",
       "            0.08672228, -0.03534925],\n",
       "          [-0.1177333 ,  0.09503537, -0.07890397, ...,  0.11153105,\n",
       "           -0.04106846,  0.03181314]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.02700698,  0.04020642,  0.00674947, ...,  0.00595032,\n",
       "           -0.01070388,  0.0197485 ],\n",
       "          [ 0.04280886, -0.01172355,  0.00478348, ..., -0.03846371,\n",
       "           -0.03034911, -0.04372284],\n",
       "          [ 0.03119953, -0.04011431,  0.02355546, ..., -0.03875683,\n",
       "           -0.00207732, -0.02862718],\n",
       "          ...,\n",
       "          [ 0.04332664, -0.00993644, -0.0410888 , ..., -0.01740349,\n",
       "           -0.01189305, -0.00527539],\n",
       "          [ 0.04923419,  0.02998776, -0.01029834, ..., -0.02852333,\n",
       "            0.01673412,  0.02181892],\n",
       "          [ 0.04720947, -0.00326831,  0.02152163, ..., -0.04969732,\n",
       "           -0.03244077, -0.02922157]],\n",
       " \n",
       "         [[ 0.03106692, -0.02053468,  0.03742464, ...,  0.03471351,\n",
       "            0.04256364,  0.02800894],\n",
       "          [ 0.01206019,  0.01481622,  0.02541111, ...,  0.0211527 ,\n",
       "           -0.02534599, -0.03247007],\n",
       "          [ 0.04569591,  0.00780565,  0.01161629, ..., -0.03550801,\n",
       "           -0.00257532,  0.02218587],\n",
       "          ...,\n",
       "          [ 0.03026138, -0.00196626, -0.04714127, ...,  0.04710475,\n",
       "            0.01036958,  0.01609673],\n",
       "          [-0.03599272, -0.00183299,  0.04981131, ...,  0.00497498,\n",
       "            0.01255242,  0.04144583],\n",
       "          [-0.050609  ,  0.05096329, -0.04766244, ...,  0.02767407,\n",
       "            0.02204708,  0.00576231]],\n",
       " \n",
       "         [[ 0.05065697,  0.02755563, -0.03136358, ..., -0.04013642,\n",
       "           -0.02048725,  0.01175742],\n",
       "          [-0.01965867, -0.00608012,  0.00623034, ...,  0.00537562,\n",
       "           -0.04450913,  0.04066151],\n",
       "          [-0.05016848,  0.02932838, -0.04255235, ...,  0.02783122,\n",
       "           -0.00609964, -0.00235422],\n",
       "          ...,\n",
       "          [-0.02971519, -0.00757002,  0.02639374, ...,  0.02472817,\n",
       "           -0.03388099, -0.02127003],\n",
       "          [ 0.0358091 , -0.03425718,  0.02492507, ..., -0.02505607,\n",
       "           -0.04075655,  0.00859962],\n",
       "          [-0.02613888,  0.00488054,  0.02696816, ..., -0.03007232,\n",
       "            0.01126608, -0.0150204 ]]],\n",
       " \n",
       " \n",
       "        [[[ 0.00541739, -0.02613055, -0.03354074, ..., -0.03741456,\n",
       "           -0.01003365, -0.044272  ],\n",
       "          [-0.02785711,  0.04559147, -0.03590014, ...,  0.01448085,\n",
       "            0.00808308, -0.05035308],\n",
       "          [-0.01879031, -0.02399903,  0.01425251, ..., -0.02239464,\n",
       "           -0.03972888,  0.00543518],\n",
       "          ...,\n",
       "          [-0.0198903 ,  0.02993234, -0.04583782, ...,  0.02722413,\n",
       "            0.00557916, -0.02229224],\n",
       "          [-0.01519651,  0.00835051,  0.00788551, ...,  0.0200302 ,\n",
       "            0.03864322, -0.02300195],\n",
       "          [ 0.01278708, -0.02638575,  0.0081462 , ...,  0.02091896,\n",
       "            0.0316152 ,  0.04245614]],\n",
       " \n",
       "         [[ 0.000717  , -0.03635799, -0.02572598, ...,  0.03898487,\n",
       "            0.02894747,  0.02396712],\n",
       "          [ 0.01530084, -0.03650623,  0.02314863, ..., -0.00123559,\n",
       "            0.00382673, -0.02917641],\n",
       "          [ 0.04653876, -0.01204682,  0.02751091, ..., -0.00252534,\n",
       "            0.00336134,  0.03748757],\n",
       "          ...,\n",
       "          [ 0.01235683, -0.03862432, -0.010218  , ..., -0.018471  ,\n",
       "           -0.0175818 , -0.03731863],\n",
       "          [ 0.01719543, -0.04219693, -0.00932253, ...,  0.02833502,\n",
       "           -0.04803744, -0.04413734],\n",
       "          [-0.01349686,  0.00243188,  0.00393797, ..., -0.01913575,\n",
       "            0.00939523,  0.03252546]],\n",
       " \n",
       "         [[ 0.00517007,  0.0112312 , -0.04962921, ...,  0.03233153,\n",
       "           -0.03112264,  0.00985699],\n",
       "          [-0.0056969 , -0.0216347 ,  0.01227919, ..., -0.00301681,\n",
       "           -0.00209732, -0.00321868],\n",
       "          [-0.03213596, -0.02886145,  0.00569973, ..., -0.00787838,\n",
       "           -0.04895   , -0.00263214],\n",
       "          ...,\n",
       "          [ 0.00791648,  0.00286666, -0.01315631, ..., -0.04447896,\n",
       "            0.02000831, -0.01951512],\n",
       "          [ 0.01710726,  0.03205887, -0.04776208, ...,  0.01545382,\n",
       "            0.04679926,  0.03005699],\n",
       "          [ 0.01440618,  0.02283999,  0.00669474, ...,  0.02700543,\n",
       "           -0.02467401, -0.00699628]]],\n",
       " \n",
       " \n",
       "        [[[-0.02754331,  0.0225953 , -0.05071376, ..., -0.01975875,\n",
       "           -0.00273367,  0.00784133],\n",
       "          [-0.04839532, -0.01280942, -0.02069733, ...,  0.02326071,\n",
       "           -0.0088353 , -0.04293972],\n",
       "          [-0.032347  ,  0.04537868, -0.04424339, ..., -0.04219944,\n",
       "            0.02939846,  0.03152659],\n",
       "          ...,\n",
       "          [ 0.0046795 , -0.04845045, -0.01162255, ...,  0.03943212,\n",
       "            0.01517982, -0.0492271 ],\n",
       "          [ 0.04156617,  0.04761124, -0.00044213, ..., -0.0399316 ,\n",
       "            0.03997948, -0.03071627],\n",
       "          [-0.04146226,  0.01216432,  0.04769072, ..., -0.00034877,\n",
       "            0.00123854, -0.03245085]],\n",
       " \n",
       "         [[-0.01628809, -0.03354947, -0.01937456, ...,  0.00610071,\n",
       "           -0.02993947,  0.04479983],\n",
       "          [-0.00567   ,  0.03088129, -0.01817499, ...,  0.01763265,\n",
       "            0.00344026, -0.0475315 ],\n",
       "          [-0.00274392, -0.01552611,  0.04395014, ...,  0.04487418,\n",
       "            0.03193583,  0.02667641],\n",
       "          ...,\n",
       "          [ 0.03804742,  0.03028025,  0.02205057, ..., -0.01514596,\n",
       "            0.04848348,  0.03981551],\n",
       "          [ 0.04894409, -0.04649233, -0.01960135, ...,  0.03816027,\n",
       "            0.00144073, -0.03417973],\n",
       "          [ 0.01927141,  0.02128293, -0.03553879, ..., -0.03799712,\n",
       "            0.04834233, -0.02332496]],\n",
       " \n",
       "         [[ 0.02160349,  0.04762585,  0.04280484, ..., -0.04512278,\n",
       "           -0.00793637, -0.03909974],\n",
       "          [-0.03982442,  0.0327545 , -0.02772089, ...,  0.03770119,\n",
       "            0.0439874 ,  0.03392577],\n",
       "          [-0.01748112, -0.03753292,  0.04345541, ..., -0.02815782,\n",
       "            0.04537582,  0.03642973],\n",
       "          ...,\n",
       "          [-0.01065464,  0.02711207,  0.02335785, ..., -0.03233089,\n",
       "            0.03518348, -0.04351366],\n",
       "          [ 0.04508819, -0.01044924,  0.0016709 , ..., -0.0289362 ,\n",
       "            0.04129677, -0.04646707],\n",
       "          [-0.02053444,  0.03585458, -0.01982773, ...,  0.01289428,\n",
       "           -0.03998232, -0.02664422]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.04377734,  0.07157903, -0.07747518, ..., -0.04486521,\n",
       "            0.02845842,  0.04684369],\n",
       "          [ 0.05669617,  0.00044677, -0.06955987, ..., -0.02834529,\n",
       "           -0.04458178,  0.03303375],\n",
       "          [-0.02143576,  0.03311914,  0.05457652, ...,  0.03451437,\n",
       "            0.06144425,  0.08822835],\n",
       "          ...,\n",
       "          [ 0.07902584, -0.03255043, -0.06125749, ...,  0.03223751,\n",
       "           -0.07288457, -0.07187298],\n",
       "          [-0.05569698,  0.01402293, -0.0745559 , ..., -0.04573249,\n",
       "           -0.01421959, -0.03829351],\n",
       "          [-0.01432186,  0.06248521,  0.06912582, ..., -0.04432177,\n",
       "           -0.06019596,  0.04306782]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([[[[-0.07062583,  0.05055876,  0.02874635, ...,  0.04589307,\n",
       "           -0.09125993, -0.06537682],\n",
       "          [-0.01259771,  0.04259829,  0.00589604, ..., -0.08449113,\n",
       "            0.019941  ,  0.02951375],\n",
       "          [-0.07623963,  0.0410094 , -0.02411106, ..., -0.08880078,\n",
       "           -0.00177238,  0.04535112],\n",
       "          ...,\n",
       "          [-0.03237627, -0.0210792 , -0.00118344, ...,  0.06803969,\n",
       "            0.01445063,  0.04300909],\n",
       "          [ 0.02858579, -0.00529525,  0.08992736, ...,  0.00799354,\n",
       "           -0.05011251, -0.05083869],\n",
       "          [ 0.03738523, -0.02584831,  0.04537211, ..., -0.05308675,\n",
       "           -0.0668412 , -0.0821492 ]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[ 0.01109593,  0.08397539, -0.08701335, ...,  0.07610419,\n",
       "            0.03732806,  0.0089002 ],\n",
       "          [ 0.00569243, -0.01269552,  0.01817696, ...,  0.02431587,\n",
       "            0.06597236,  0.08035743],\n",
       "          [ 0.05027655, -0.03536221, -0.00766128, ...,  0.05178897,\n",
       "            0.03662942,  0.07585707],\n",
       "          ...,\n",
       "          [-0.04329906,  0.04327835,  0.09095882, ...,  0.03528993,\n",
       "            0.07053263, -0.03908498],\n",
       "          [ 0.01146125, -0.0788406 ,  0.07011843, ..., -0.04176399,\n",
       "           -0.09387641,  0.07406415],\n",
       "          [ 0.04022403, -0.05662221,  0.03107226, ..., -0.07769506,\n",
       "            0.04598176, -0.03146319]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.0373833 , -0.01569839,  0.01336952, ...,  0.04219062,\n",
       "           -0.04835973,  0.03201042],\n",
       "          [ 0.02263758,  0.02059389, -0.0062143 , ..., -0.04718992,\n",
       "            0.02307616,  0.00310213],\n",
       "          [-0.01712578, -0.02748544, -0.03306885, ...,  0.02688377,\n",
       "            0.00668355,  0.02838681],\n",
       "          ...,\n",
       "          [-0.00070361, -0.02897774, -0.04964711, ...,  0.01321471,\n",
       "            0.01750566, -0.00356931],\n",
       "          [ 0.04497243,  0.04388626,  0.01594764, ..., -0.04353166,\n",
       "            0.04740346, -0.02641873],\n",
       "          [ 0.00433989,  0.01843684,  0.00955391, ..., -0.03442357,\n",
       "           -0.03247489,  0.04166628]],\n",
       " \n",
       "         [[ 0.04550091,  0.04809922,  0.03873177, ..., -0.03626863,\n",
       "           -0.01885594,  0.01934866],\n",
       "          [ 0.00044309, -0.02791305, -0.03765809, ..., -0.009198  ,\n",
       "           -0.01432532, -0.02709683],\n",
       "          [-0.0259981 , -0.00923955, -0.03724712, ..., -0.03587947,\n",
       "           -0.04972949,  0.00620368],\n",
       "          ...,\n",
       "          [ 0.01963027,  0.02812549, -0.00322839, ...,  0.01048022,\n",
       "            0.01423625,  0.02999822],\n",
       "          [-0.04406306, -0.0299323 ,  0.00461041, ...,  0.0369682 ,\n",
       "           -0.00858747, -0.01925053],\n",
       "          [-0.03179804, -0.04797906, -0.03899522, ...,  0.00376028,\n",
       "            0.02158099,  0.02638286]],\n",
       " \n",
       "         [[ 0.04591709,  0.01537126,  0.02119563, ..., -0.01397686,\n",
       "            0.00311267, -0.00568317],\n",
       "          [-0.03500009,  0.00080736, -0.00230349, ..., -0.0197557 ,\n",
       "            0.00442918,  0.03074203],\n",
       "          [-0.0182013 , -0.01411479,  0.002054  , ..., -0.00349331,\n",
       "            0.01999816,  0.03024297],\n",
       "          ...,\n",
       "          [-0.04020175,  0.02879446,  0.02329273, ..., -0.03193119,\n",
       "            0.00635595, -0.03473459],\n",
       "          [ 0.0394138 ,  0.01802691,  0.01552229, ...,  0.001886  ,\n",
       "            0.03837606, -0.00715202],\n",
       "          [-0.04330393,  0.00038223,  0.04154088, ...,  0.0093257 ,\n",
       "           -0.01830677, -0.01958296]]],\n",
       " \n",
       " \n",
       "        [[[ 0.02074222, -0.02244988, -0.0368062 , ...,  0.04203174,\n",
       "           -0.00377674, -0.0410165 ],\n",
       "          [ 0.02252496,  0.02836562, -0.01540636, ...,  0.0150633 ,\n",
       "            0.02421587,  0.03494479],\n",
       "          [ 0.04094737,  0.00018082,  0.00943775, ..., -0.00069154,\n",
       "            0.02712679,  0.01158026],\n",
       "          ...,\n",
       "          [-0.0445145 ,  0.040316  ,  0.03360272, ...,  0.00833651,\n",
       "           -0.04765071, -0.02985492],\n",
       "          [ 0.03062221,  0.05040826, -0.04278146, ...,  0.02313557,\n",
       "            0.00267839,  0.00213797],\n",
       "          [ 0.00530027, -0.00207626,  0.01955087, ..., -0.02882155,\n",
       "            0.00458986, -0.03437205]],\n",
       " \n",
       "         [[-0.03804287, -0.00047851, -0.0499318 , ..., -0.02399865,\n",
       "            0.00565967, -0.0257274 ],\n",
       "          [ 0.04586893, -0.01728661,  0.04541363, ..., -0.04826269,\n",
       "           -0.0391272 , -0.00824617],\n",
       "          [ 0.03706206, -0.01620774, -0.01244462, ...,  0.02943326,\n",
       "            0.03662955,  0.02756372],\n",
       "          ...,\n",
       "          [ 0.04401452, -0.01369792,  0.02534555, ...,  0.01698719,\n",
       "            0.02542523,  0.03361007],\n",
       "          [ 0.02493927,  0.04427239, -0.04451101, ..., -0.04398334,\n",
       "           -0.02926049,  0.03222553],\n",
       "          [ 0.0077251 , -0.01327087,  0.04594227, ..., -0.04395242,\n",
       "           -0.03868187, -0.04410665]],\n",
       " \n",
       "         [[-0.02470428,  0.03868629,  0.01765797, ...,  0.03565889,\n",
       "            0.00200537, -0.02063295],\n",
       "          [ 0.02005032, -0.00902321,  0.05036724, ...,  0.02441027,\n",
       "            0.03324954,  0.03122532],\n",
       "          [ 0.02927312, -0.00577917, -0.00494043, ...,  0.036952  ,\n",
       "           -0.01996628, -0.02640017],\n",
       "          ...,\n",
       "          [-0.01876448,  0.0099066 , -0.04811972, ..., -0.008993  ,\n",
       "           -0.03558221,  0.04384355],\n",
       "          [ 0.01609344, -0.00291399, -0.0237862 , ...,  0.02340867,\n",
       "           -0.04177325,  0.00192736],\n",
       "          [ 0.04033969, -0.02468577, -0.01593396, ...,  0.01071117,\n",
       "           -0.01042778,  0.02936385]]],\n",
       " \n",
       " \n",
       "        [[[-0.04313213, -0.03007447, -0.00768269, ...,  0.03214862,\n",
       "           -0.03670469,  0.00404184],\n",
       "          [-0.043334  ,  0.04579791,  0.00375308, ...,  0.04869118,\n",
       "           -0.02738482, -0.01716155],\n",
       "          [-0.01147131, -0.00074981,  0.01386311, ..., -0.03224751,\n",
       "            0.05041383,  0.02392676],\n",
       "          ...,\n",
       "          [-0.0210254 ,  0.00149122, -0.01789556, ..., -0.02463931,\n",
       "            0.00258308,  0.00332474],\n",
       "          [ 0.02010354, -0.01497932, -0.04802258, ...,  0.01037883,\n",
       "            0.03561182,  0.01200292],\n",
       "          [-0.02504467, -0.01946384, -0.02200023, ...,  0.02799456,\n",
       "            0.00085198, -0.00875775]],\n",
       " \n",
       "         [[-0.01114458,  0.01941313,  0.03104351, ...,  0.02461832,\n",
       "            0.02511375,  0.00029047],\n",
       "          [ 0.01906151,  0.0298269 ,  0.00568186, ..., -0.03030239,\n",
       "           -0.0119158 , -0.02234224],\n",
       "          [ 0.05009546,  0.01998167, -0.04173722, ..., -0.02890491,\n",
       "           -0.00708135, -0.03093818],\n",
       "          ...,\n",
       "          [ 0.05032678, -0.00223771,  0.02992924, ..., -0.0178438 ,\n",
       "            0.04926913,  0.03826977],\n",
       "          [-0.00605153,  0.00145113,  0.04608487, ...,  0.02048104,\n",
       "            0.01940814, -0.02062357],\n",
       "          [-0.03160057,  0.01359256,  0.00633525, ..., -0.0476975 ,\n",
       "           -0.02064349, -0.03446831]],\n",
       " \n",
       "         [[-0.02358658,  0.04829037,  0.00413378, ..., -0.03394087,\n",
       "            0.00483179, -0.04295551],\n",
       "          [-0.00882595,  0.04416121,  0.00628583, ...,  0.03583407,\n",
       "           -0.04693512,  0.00355792],\n",
       "          [ 0.04153749,  0.02778252, -0.00819432, ..., -0.0068546 ,\n",
       "            0.02893847,  0.05087873],\n",
       "          ...,\n",
       "          [ 0.03227375, -0.04139397,  0.02474689, ...,  0.0333812 ,\n",
       "            0.01514103, -0.03075815],\n",
       "          [ 0.03567471,  0.04558716,  0.01668672, ...,  0.00611525,\n",
       "           -0.04598801,  0.00537448],\n",
       "          [-0.0426662 ,  0.02226046, -0.01169273, ..., -0.02899648,\n",
       "           -0.02338066,  0.01530802]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.03405513, -0.01559215, -0.00707537, ..., -0.06628781,\n",
       "            0.06378104,  0.04810327],\n",
       "          [ 0.06843525,  0.01731934,  0.02602765, ...,  0.09566469,\n",
       "            0.08260584, -0.01132706],\n",
       "          [ 0.03310542,  0.07934438, -0.09289888, ..., -0.00044943,\n",
       "            0.00421689, -0.00438987],\n",
       "          ...,\n",
       "          [ 0.03460225, -0.09444704, -0.00979064, ..., -0.03857201,\n",
       "            0.00469036, -0.05121862],\n",
       "          [-0.03914068,  0.09590502,  0.05427663, ..., -0.01334787,\n",
       "            0.02215672,  0.04021209],\n",
       "          [ 0.05641396,  0.09630933, -0.01686241, ...,  0.09114781,\n",
       "            0.02080003,  0.02711386]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[-0.07357819, -0.07450828,  0.01608167, ..., -0.05010907,\n",
       "           -0.0918898 , -0.00608882],\n",
       "          [-0.05591787,  0.04402491,  0.09241779, ...,  0.03204814,\n",
       "            0.08800377, -0.09145796],\n",
       "          [ 0.01708391,  0.07920706, -0.02852806, ..., -0.02296543,\n",
       "            0.00281523,  0.04895672],\n",
       "          ...,\n",
       "          [ 0.05508913,  0.09172504,  0.03890029, ..., -0.07054301,\n",
       "           -0.03143748,  0.09399077],\n",
       "          [ 0.08557256,  0.08857517,  0.0332268 , ...,  0.00194348,\n",
       "           -0.00134277, -0.00182594],\n",
       "          [-0.01988532,  0.00460777, -0.00777599, ...,  0.04407136,\n",
       "           -0.09537276, -0.08795322]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 7.69904628e-03,  3.03787366e-02, -4.31559905e-02, ...,\n",
       "           -2.41022054e-02, -4.74465787e-02, -4.70313281e-02],\n",
       "          [ 3.92071009e-02, -4.59563248e-02,  1.18823722e-02, ...,\n",
       "            9.04838741e-03,  1.30613074e-02, -1.59881413e-02],\n",
       "          [-4.18614708e-02,  2.04984099e-04,  1.16807967e-03, ...,\n",
       "           -2.04037577e-02,  4.49458361e-02,  2.78903320e-02],\n",
       "          ...,\n",
       "          [ 7.25796446e-03, -4.84408364e-02, -3.55168059e-02, ...,\n",
       "           -3.24042216e-02,  7.69383833e-03, -4.62325439e-02],\n",
       "          [ 2.68370286e-03, -2.50936132e-02, -2.10717116e-02, ...,\n",
       "           -3.13156620e-02,  4.15718332e-02,  4.44170237e-02],\n",
       "          [ 4.61475104e-02,  2.42168158e-02, -4.29712608e-02, ...,\n",
       "           -4.75977398e-02,  3.97229344e-02,  3.97406965e-02]],\n",
       " \n",
       "         [[ 4.14460748e-02, -3.10574472e-03,  1.53268427e-02, ...,\n",
       "            8.15333799e-03, -2.64847092e-02, -4.50715832e-02],\n",
       "          [-2.28179563e-02, -3.21824960e-02, -4.72102053e-02, ...,\n",
       "           -3.65883298e-02, -7.79420137e-03,  8.46106187e-03],\n",
       "          [ 1.58367008e-02,  5.30129671e-03,  1.01731531e-02, ...,\n",
       "            3.03675458e-02,  2.92721614e-02,  3.56357843e-02],\n",
       "          ...,\n",
       "          [ 3.23377028e-02,  4.39958349e-02, -3.81956920e-02, ...,\n",
       "            2.84849629e-02, -1.78764760e-03,  2.25093216e-02],\n",
       "          [-3.45893726e-02,  4.06271964e-03,  5.45525178e-03, ...,\n",
       "           -3.91537733e-02, -4.02402133e-04, -1.71227641e-02],\n",
       "          [ 7.07925856e-03,  9.21849161e-04,  7.05739483e-03, ...,\n",
       "            3.04644257e-02,  3.10246125e-02, -4.67801094e-02]],\n",
       " \n",
       "         [[ 1.20081082e-02,  1.06993318e-03, -2.79933754e-02, ...,\n",
       "            6.21905923e-03, -2.22574919e-03, -2.03244053e-02],\n",
       "          [-4.86751422e-02, -1.19089708e-02, -3.31053212e-02, ...,\n",
       "            1.94204524e-02,  1.34206265e-02, -2.47576032e-02],\n",
       "          [ 4.02583331e-02, -2.39026546e-03,  2.02880055e-02, ...,\n",
       "           -1.46308169e-02,  4.65140939e-02, -9.61152464e-03],\n",
       "          ...,\n",
       "          [-1.81127228e-02,  2.41866037e-02, -2.63053477e-02, ...,\n",
       "            2.54005194e-03, -3.92674953e-02,  4.13156599e-02],\n",
       "          [-2.99809109e-02,  4.47473675e-03, -5.00083119e-02, ...,\n",
       "           -3.00009847e-02,  2.16854289e-02,  1.86813250e-03],\n",
       "          [ 1.22239590e-02, -2.18235794e-02, -2.81914379e-02, ...,\n",
       "           -5.41371480e-03,  5.02896681e-03,  8.41996074e-03]]],\n",
       " \n",
       " \n",
       "        [[[ 3.86776179e-02, -5.20753860e-03, -4.29364033e-02, ...,\n",
       "           -1.32244527e-02,  2.53175870e-02, -4.72198278e-02],\n",
       "          [ 4.72482368e-02, -3.99551354e-02,  2.08623111e-02, ...,\n",
       "            4.92760465e-02, -4.20728698e-02, -3.41404416e-02],\n",
       "          [-1.57811977e-02, -7.88211823e-03, -2.33982857e-02, ...,\n",
       "           -1.25287808e-02,  1.00304931e-03,  4.88724783e-02],\n",
       "          ...,\n",
       "          [ 3.98408547e-02, -4.43114266e-02,  3.04034576e-02, ...,\n",
       "           -2.33406890e-02,  3.43002975e-02, -2.96522491e-02],\n",
       "          [ 2.13117898e-02, -3.06451321e-03,  2.54179165e-03, ...,\n",
       "            9.56490263e-03,  4.67806086e-02,  3.82492840e-02],\n",
       "          [-3.86331715e-02, -4.84891720e-02,  2.53122002e-02, ...,\n",
       "            2.50931755e-02,  1.84446275e-02, -4.03754413e-04]],\n",
       " \n",
       "         [[ 1.90400928e-02, -1.51746608e-02,  2.18900740e-02, ...,\n",
       "            3.37663591e-02,  2.74960250e-02,  4.02280241e-02],\n",
       "          [ 3.02234292e-02, -2.53818557e-02,  2.67813131e-02, ...,\n",
       "           -3.53077203e-02, -9.34862345e-03,  5.01765311e-03],\n",
       "          [-1.79584697e-03, -2.61635315e-02, -1.25789940e-02, ...,\n",
       "            1.01044774e-03,  2.32284144e-02, -2.90913507e-02],\n",
       "          ...,\n",
       "          [ 4.14312780e-02,  1.20315850e-02,  5.04832938e-02, ...,\n",
       "           -2.17364915e-02, -1.91845000e-04, -4.98748682e-02],\n",
       "          [-2.77160220e-02, -9.04439390e-03,  4.68148962e-02, ...,\n",
       "            4.10506278e-02, -5.77862188e-03,  4.00758535e-02],\n",
       "          [-1.69812739e-02, -3.71457748e-02,  5.95707074e-03, ...,\n",
       "           -3.89154628e-03,  2.40731761e-02,  1.38973966e-02]],\n",
       " \n",
       "         [[ 2.73918882e-02,  3.90107259e-02, -2.06026100e-02, ...,\n",
       "            3.12723517e-02,  3.71271223e-02,  8.68641213e-03],\n",
       "          [-4.15791906e-02,  1.14619024e-02,  2.80044004e-02, ...,\n",
       "           -4.54769060e-02,  3.32836807e-02, -6.23504445e-03],\n",
       "          [ 1.42393187e-02, -9.65480134e-03, -3.37118581e-02, ...,\n",
       "           -4.85364161e-02, -3.59772444e-02,  1.32701099e-02],\n",
       "          ...,\n",
       "          [ 8.44532996e-03, -1.26772411e-02,  1.49869919e-03, ...,\n",
       "            1.17900148e-02,  2.90311128e-02, -6.32452965e-03],\n",
       "          [ 1.11424774e-02, -4.59499955e-02,  3.08653340e-02, ...,\n",
       "            2.81106755e-02,  1.06245279e-02, -1.08416565e-02],\n",
       "          [-4.55783643e-02, -7.17588514e-03,  2.12634727e-03, ...,\n",
       "            4.68424559e-02, -5.96918911e-03, -4.30712104e-02]]],\n",
       " \n",
       " \n",
       "        [[[-3.80722582e-02,  4.84087989e-02, -4.82329912e-02, ...,\n",
       "           -2.63735671e-02, -1.43752210e-02,  2.67870799e-02],\n",
       "          [-6.39509782e-03, -1.26614608e-02,  1.50436983e-02, ...,\n",
       "           -1.82398297e-02,  2.72917897e-02,  3.89956757e-02],\n",
       "          [-3.14817764e-02, -3.06073520e-02, -1.95105001e-02, ...,\n",
       "            2.05694064e-02, -1.12513825e-02,  4.65924889e-02],\n",
       "          ...,\n",
       "          [ 3.74487489e-02,  4.02900502e-02, -3.80116552e-02, ...,\n",
       "            3.43921036e-03, -2.37805527e-02,  2.51914561e-02],\n",
       "          [-4.56708297e-02, -3.64103913e-02, -3.04585043e-02, ...,\n",
       "            4.27061021e-02, -1.74225867e-02,  2.41911784e-02],\n",
       "          [ 6.29349425e-03,  3.31835970e-02, -1.08478591e-03, ...,\n",
       "           -4.20446768e-02, -5.01607992e-02, -7.76995346e-03]],\n",
       " \n",
       "         [[-3.00045498e-02,  6.20399415e-03,  4.67717275e-02, ...,\n",
       "           -3.47089544e-02, -2.69575100e-02,  3.47366333e-02],\n",
       "          [ 1.25948116e-02, -1.76114403e-02, -3.44830565e-02, ...,\n",
       "            1.30676851e-03, -4.63181287e-02, -1.23563558e-02],\n",
       "          [-4.53828461e-02, -1.92354321e-02,  2.49127075e-02, ...,\n",
       "           -6.27148896e-04,  3.43755186e-02, -1.30418651e-02],\n",
       "          ...,\n",
       "          [-2.44642999e-02, -1.66454613e-02, -6.09853491e-03, ...,\n",
       "           -1.68538056e-02,  2.07225159e-02,  3.41665372e-02],\n",
       "          [ 5.89458272e-03,  1.30987242e-02,  4.01692092e-02, ...,\n",
       "           -1.55219808e-03,  4.65405136e-02, -3.09453439e-02],\n",
       "          [-2.48720814e-02,  1.20919198e-03, -1.94147825e-02, ...,\n",
       "           -1.99620072e-02, -7.34457746e-03, -3.85106504e-02]],\n",
       " \n",
       "         [[-4.00352180e-02, -3.23074088e-02,  4.76130322e-02, ...,\n",
       "           -4.23800536e-02,  3.30713764e-02, -8.32215324e-03],\n",
       "          [ 3.10942009e-02,  9.30470228e-03,  4.78879288e-02, ...,\n",
       "           -1.03013776e-02, -4.16180380e-02, -2.75770035e-02],\n",
       "          [ 2.74993107e-02,  3.44900191e-02, -4.37801071e-02, ...,\n",
       "            2.37531140e-02, -3.44270170e-02, -9.97558236e-05],\n",
       "          ...,\n",
       "          [ 7.67035410e-03,  5.09640351e-02, -6.34519011e-03, ...,\n",
       "           -3.18700038e-02,  6.14944100e-03, -5.01626842e-02],\n",
       "          [-1.18374079e-03, -3.86705697e-02, -3.87280807e-03, ...,\n",
       "           -2.34823711e-02, -4.99974713e-02, -2.16512755e-02],\n",
       "          [ 1.04531571e-02,  4.63862270e-02,  2.82680020e-02, ...,\n",
       "           -3.90989333e-03,  2.88432091e-03,  1.91848353e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.02535037, -0.00348959,  0.01561859, ...,  0.05637585,\n",
       "            0.04500942,  0.03296985],\n",
       "          [-0.01284701,  0.02730121, -0.01396669, ...,  0.08232786,\n",
       "            0.04689331,  0.04155505],\n",
       "          [ 0.08885074, -0.02465863, -0.01182975, ..., -0.01802534,\n",
       "           -0.01763153, -0.00604574],\n",
       "          ...,\n",
       "          [-0.07959104,  0.08877084,  0.01537781, ..., -0.03291689,\n",
       "            0.09642097,  0.06482606],\n",
       "          [ 0.06987694,  0.03737973,  0.06724112, ..., -0.06042972,\n",
       "            0.02033467,  0.00412179],\n",
       "          [-0.07741128,  0.01036825,  0.02635412, ..., -0.0121512 ,\n",
       "           -0.07918121,  0.09666729]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[-0.05182725,  0.0178996 , -0.07724978, ..., -0.07667642,\n",
       "           -0.08556817,  0.01704881],\n",
       "          [ 0.03140765,  0.06746356,  0.07864325, ..., -0.03420229,\n",
       "            0.03058025,  0.05019352],\n",
       "          [ 0.07392725,  0.06476717,  0.08754693, ...,  0.07910776,\n",
       "            0.06126264, -0.08744376],\n",
       "          ...,\n",
       "          [ 0.04184604, -0.09571239, -0.00768015, ...,  0.04622175,\n",
       "           -0.07183827, -0.01992452],\n",
       "          [-0.03028924,  0.07372983, -0.06998084, ..., -0.0425899 ,\n",
       "           -0.08563399, -0.08655909],\n",
       "          [-0.06049627,  0.03916065, -0.04315792, ...,  0.07203086,\n",
       "           -0.06458443,  0.03815554]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.01696042, -0.03174102, -0.03277646, ..., -0.04999281,\n",
       "           -0.01474503,  0.02398299],\n",
       "          [ 0.03109231, -0.02592334,  0.03118934, ..., -0.05020377,\n",
       "            0.00057263,  0.01489205],\n",
       "          [-0.03510398,  0.01319584,  0.04249743, ...,  0.02745411,\n",
       "           -0.04529556,  0.01124417],\n",
       "          ...,\n",
       "          [ 0.0181442 , -0.00196154,  0.01829343, ...,  0.01206874,\n",
       "           -0.03257115,  0.01390501],\n",
       "          [-0.02555492,  0.01066749, -0.02580562, ..., -0.01458903,\n",
       "            0.02769345, -0.03175057],\n",
       "          [-0.049479  ,  0.05011979,  0.03253821, ...,  0.0142206 ,\n",
       "            0.04863395, -0.02634038]],\n",
       " \n",
       "         [[ 0.04167794, -0.01485829,  0.03434107, ...,  0.02511808,\n",
       "            0.00055731, -0.01502896],\n",
       "          [-0.03363619, -0.00076435, -0.01158064, ..., -0.01891345,\n",
       "            0.05089989, -0.02658146],\n",
       "          [-0.01572597,  0.02449114, -0.03638499, ..., -0.02192615,\n",
       "           -0.04547085, -0.04682932],\n",
       "          ...,\n",
       "          [-0.04705904,  0.00520641,  0.01161706, ...,  0.00254777,\n",
       "           -0.02334446, -0.01766001],\n",
       "          [ 0.03965877, -0.01863199, -0.02734328, ...,  0.04709551,\n",
       "            0.02327062, -0.00349872],\n",
       "          [ 0.02443853,  0.04226781,  0.04619843, ..., -0.00356527,\n",
       "           -0.00946512, -0.03870047]],\n",
       " \n",
       "         [[ 0.03715034,  0.04901641,  0.01946549, ...,  0.04843803,\n",
       "            0.02468345, -0.01307246],\n",
       "          [ 0.03986288, -0.00041738,  0.00221699, ..., -0.03037505,\n",
       "           -0.03754101, -0.03339332],\n",
       "          [-0.00896708, -0.03683707, -0.04834926, ...,  0.03882812,\n",
       "            0.02462563,  0.01020732],\n",
       "          ...,\n",
       "          [ 0.02464662, -0.04183876, -0.04954638, ..., -0.02005327,\n",
       "           -0.0408223 ,  0.01646613],\n",
       "          [-0.02556321,  0.02585529,  0.02804727, ..., -0.04495157,\n",
       "           -0.02895415, -0.02930325],\n",
       "          [-0.0150499 ,  0.03822418,  0.03180326, ..., -0.01483354,\n",
       "            0.00880603,  0.02253738]]],\n",
       " \n",
       " \n",
       "        [[[-0.02447948,  0.02301339,  0.04401063, ..., -0.04382634,\n",
       "            0.03052682,  0.02852353],\n",
       "          [ 0.03963803,  0.0241677 ,  0.01440151, ...,  0.01314163,\n",
       "            0.03294579, -0.0191686 ],\n",
       "          [-0.0295915 , -0.0314549 , -0.00275978, ...,  0.00364556,\n",
       "           -0.04407974, -0.02268565],\n",
       "          ...,\n",
       "          [ 0.02158525,  0.04901624, -0.04089854, ...,  0.04785734,\n",
       "            0.02325744, -0.00929989],\n",
       "          [-0.02575803,  0.04187661, -0.04775509, ..., -0.04275405,\n",
       "           -0.00835666, -0.03179327],\n",
       "          [-0.04396778, -0.02067598,  0.03574596, ..., -0.04062629,\n",
       "            0.01238476, -0.04290726]],\n",
       " \n",
       "         [[ 0.04159217,  0.02861952, -0.0113261 , ..., -0.03318618,\n",
       "           -0.00732966, -0.0199895 ],\n",
       "          [-0.006888  , -0.00800429,  0.04640393, ..., -0.04952445,\n",
       "           -0.03733033,  0.00641116],\n",
       "          [-0.00600996,  0.01797716,  0.04247274, ..., -0.04425899,\n",
       "           -0.04533686, -0.03372993],\n",
       "          ...,\n",
       "          [ 0.02773429,  0.02189006, -0.03041029, ...,  0.0069586 ,\n",
       "            0.04289928,  0.01770253],\n",
       "          [ 0.042718  ,  0.00018613,  0.04612102, ...,  0.05084335,\n",
       "            0.02003121,  0.02634209],\n",
       "          [-0.03481524,  0.02527612, -0.02379474, ...,  0.00953582,\n",
       "           -0.0030007 ,  0.03388604]],\n",
       " \n",
       "         [[-0.02132698,  0.04761936,  0.040424  , ...,  0.00074926,\n",
       "           -0.02786652,  0.02701166],\n",
       "          [ 0.03829975,  0.00563841, -0.01387354, ..., -0.00578183,\n",
       "           -0.00995344,  0.01390231],\n",
       "          [-0.00163926, -0.04155717, -0.04037288, ..., -0.02069075,\n",
       "            0.00952081, -0.03848087],\n",
       "          ...,\n",
       "          [ 0.04322204, -0.04010785,  0.02380881, ..., -0.04307121,\n",
       "           -0.01122891, -0.00122018],\n",
       "          [-0.02808062,  0.0425989 ,  0.00804359, ...,  0.01741265,\n",
       "           -0.0290264 ,  0.04691204],\n",
       "          [-0.02324893, -0.04201312,  0.02293952, ...,  0.01237471,\n",
       "           -0.03461195, -0.00221375]]],\n",
       " \n",
       " \n",
       "        [[[-0.02724584,  0.01916181, -0.03906543, ..., -0.03462766,\n",
       "            0.04824755,  0.03846689],\n",
       "          [ 0.01202278, -0.0092578 , -0.04979746, ...,  0.01796225,\n",
       "            0.03175876,  0.04681858],\n",
       "          [ 0.02127741, -0.02311542,  0.02594486, ..., -0.03625271,\n",
       "            0.01000227, -0.02276996],\n",
       "          ...,\n",
       "          [ 0.02635046,  0.03678454, -0.00680568, ...,  0.00724303,\n",
       "           -0.01926838, -0.00026198],\n",
       "          [ 0.02241413, -0.0380835 , -0.03322051, ...,  0.0136799 ,\n",
       "            0.0340401 ,  0.0222066 ],\n",
       "          [-0.03707399,  0.0234579 , -0.03209898, ...,  0.0354353 ,\n",
       "            0.04446911,  0.04485255]],\n",
       " \n",
       "         [[-0.00016465,  0.03328627,  0.03449044, ..., -0.03567693,\n",
       "           -0.00796574,  0.03077349],\n",
       "          [-0.03178889, -0.02853925,  0.03240357, ..., -0.00050638,\n",
       "            0.04671502,  0.03774104],\n",
       "          [-0.04663775,  0.01085094,  0.03966441, ..., -0.03668452,\n",
       "            0.0309096 ,  0.00332392],\n",
       "          ...,\n",
       "          [ 0.04660425, -0.01248789, -0.04619683, ..., -0.0209097 ,\n",
       "            0.01447264, -0.02429039],\n",
       "          [-0.01670191,  0.01145248, -0.02151294, ..., -0.00523674,\n",
       "           -0.01240737,  0.0327797 ],\n",
       "          [-0.03051134, -0.03534847,  0.02897022, ..., -0.0489684 ,\n",
       "            0.0140944 , -0.02605073]],\n",
       " \n",
       "         [[-0.01027084, -0.00685803, -0.03835158, ...,  0.02051295,\n",
       "            0.0492221 ,  0.0490863 ],\n",
       "          [-0.03216788,  0.01036951,  0.02858693, ...,  0.02602278,\n",
       "           -0.02207074, -0.02766885],\n",
       "          [ 0.03735933,  0.01713679,  0.04983354, ..., -0.04935632,\n",
       "            0.03266928, -0.02266317],\n",
       "          ...,\n",
       "          [ 0.00048139, -0.01721675, -0.03656577, ..., -0.03453124,\n",
       "           -0.04400387, -0.0383965 ],\n",
       "          [-0.04118621,  0.02954524,  0.03465767, ..., -0.00998233,\n",
       "            0.02385223,  0.03628705],\n",
       "          [-0.00243694,  0.04508086,  0.04059356, ...,  0.0151692 ,\n",
       "            0.04641696,  0.0445286 ]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.01078741, -0.09629931,  0.06132804, ..., -0.07948   ,\n",
       "            0.02347002, -0.06125491],\n",
       "          [-0.03731142,  0.04915898,  0.04428335, ..., -0.02550146,\n",
       "            0.08231731,  0.03212616],\n",
       "          [ 0.00332732,  0.01735754,  0.054278  , ...,  0.06484199,\n",
       "           -0.07328589, -0.00361586],\n",
       "          ...,\n",
       "          [ 0.07127748,  0.02160878, -0.03121594, ..., -0.08931289,\n",
       "            0.03194682,  0.02203085],\n",
       "          [-0.08393266,  0.04486693, -0.05228126, ..., -0.02996247,\n",
       "            0.01019269,  0.03455228],\n",
       "          [-0.00012692,  0.05904597, -0.03583577, ...,  0.08438139,\n",
       "           -0.03587765,  0.01282828]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[-0.01624615,  0.07102077, -0.0487364 , ..., -0.01599424,\n",
       "            0.05099054,  0.08597688],\n",
       "          [ 0.07295545,  0.06714382, -0.01327152, ...,  0.05788901,\n",
       "            0.05323055,  0.00591616],\n",
       "          [ 0.03517772, -0.00192706,  0.00975868, ...,  0.02577267,\n",
       "           -0.02547263, -0.04360226],\n",
       "          ...,\n",
       "          [ 0.02132729, -0.05364373, -0.00255273, ..., -0.08676238,\n",
       "           -0.04945437, -0.05522611],\n",
       "          [ 0.04141844, -0.01926977, -0.00754924, ..., -0.08736167,\n",
       "            0.01023863, -0.02706932],\n",
       "          [-0.0598328 ,  0.06471565, -0.08623566, ...,  0.07851353,\n",
       "            0.04701491,  0.0780386 ]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 0.01459201, -0.00888498, -0.02431213, ..., -0.02581026,\n",
       "            0.0305301 ,  0.02848092],\n",
       "          [ 0.02987566, -0.00588506,  0.00725473, ...,  0.00373959,\n",
       "            0.02090548,  0.03244337],\n",
       "          [ 0.00392039, -0.03026826,  0.00552036, ..., -0.03532528,\n",
       "            0.0212865 , -0.0340813 ],\n",
       "          ...,\n",
       "          [-0.01537701, -0.03125582, -0.01190669, ...,  0.03476262,\n",
       "            0.0141589 , -0.03446696],\n",
       "          [-0.03228083, -0.02198888,  0.01826927, ..., -0.02766202,\n",
       "            0.02661458, -0.01114747],\n",
       "          [-0.01622652, -0.02849267,  0.01372842, ..., -0.02084692,\n",
       "            0.03033797, -0.01867734]],\n",
       " \n",
       "         [[ 0.01282294, -0.02085926, -0.01769134, ..., -0.02829564,\n",
       "           -0.022577  ,  0.02569201],\n",
       "          [ 0.03540258, -0.03155755, -0.01053035, ..., -0.03194678,\n",
       "           -0.02659123,  0.01794854],\n",
       "          [ 0.02115347,  0.02082767,  0.00975789, ..., -0.03423657,\n",
       "           -0.03245938,  0.0080967 ],\n",
       "          ...,\n",
       "          [-0.00722635,  0.02814303,  0.02996817, ..., -0.01670789,\n",
       "           -0.00920349,  0.00658692],\n",
       "          [ 0.00924506,  0.0235033 ,  0.02808213, ...,  0.00466906,\n",
       "           -0.03189501,  0.0277053 ],\n",
       "          [-0.03111862,  0.00933243, -0.03090101, ..., -0.01928601,\n",
       "            0.00986622,  0.015802  ]],\n",
       " \n",
       "         [[-0.00548877, -0.01968355,  0.00055219, ..., -0.01780187,\n",
       "           -0.00281851,  0.03546473],\n",
       "          [ 0.00861712, -0.01779528,  0.01530523, ..., -0.03592424,\n",
       "           -0.03236956,  0.00346532],\n",
       "          [-0.02774371, -0.0221144 ,  0.00710347, ..., -0.02471977,\n",
       "           -0.00591627,  0.01142566],\n",
       "          ...,\n",
       "          [ 0.01957623,  0.02717748,  0.00140389, ..., -0.00440291,\n",
       "            0.00403914,  0.00437082],\n",
       "          [ 0.02272794,  0.00813786, -0.01838188, ...,  0.01667468,\n",
       "           -0.00226661, -0.02980627],\n",
       "          [-0.015199  , -0.00752401, -0.02698202, ...,  0.02317186,\n",
       "            0.03174176,  0.01034942]]],\n",
       " \n",
       " \n",
       "        [[[ 0.00223633,  0.00842172, -0.03183689, ..., -0.01930737,\n",
       "           -0.02653776, -0.03191265],\n",
       "          [ 0.02613208, -0.03388224,  0.01681099, ..., -0.02065465,\n",
       "           -0.02478092, -0.03528621],\n",
       "          [-0.03106825,  0.00932664,  0.01807685, ...,  0.01529268,\n",
       "           -0.03083158, -0.03198909],\n",
       "          ...,\n",
       "          [-0.00735402,  0.01317696,  0.00237319, ...,  0.02520537,\n",
       "            0.01220727,  0.03306396],\n",
       "          [-0.02572508,  0.01833249, -0.02342216, ..., -0.03109257,\n",
       "            0.00246134,  0.00942769],\n",
       "          [ 0.01289257, -0.01526006, -0.02307308, ..., -0.00318837,\n",
       "           -0.01465078,  0.03438332]],\n",
       " \n",
       "         [[ 0.01168028, -0.00237615, -0.00062204, ..., -0.00262923,\n",
       "            0.03563833,  0.01897725],\n",
       "          [-0.00149354,  0.01917455, -0.00748753, ...,  0.02602731,\n",
       "           -0.02445084, -0.01864585],\n",
       "          [-0.01572849, -0.01564508,  0.02991685, ..., -0.00854907,\n",
       "           -0.00923636,  0.01798388],\n",
       "          ...,\n",
       "          [ 0.01170669,  0.02231941,  0.00023535, ...,  0.01976725,\n",
       "            0.00279652, -0.0160574 ],\n",
       "          [-0.03301095,  0.01328634, -0.00419882, ...,  0.01428151,\n",
       "            0.03129528, -0.01638445],\n",
       "          [ 0.01212882,  0.02597373, -0.03017006, ..., -0.03505224,\n",
       "            0.00342476,  0.02427023]],\n",
       " \n",
       "         [[ 0.02471944,  0.03362209,  0.03322594, ...,  0.00167781,\n",
       "            0.00855641,  0.03238499],\n",
       "          [-0.01162951, -0.02352585, -0.03103508, ..., -0.03190894,\n",
       "           -0.02008834, -0.0143368 ],\n",
       "          [-0.00685515,  0.0293273 ,  0.01136909, ...,  0.02782826,\n",
       "            0.01040517, -0.01136653],\n",
       "          ...,\n",
       "          [ 0.00025385,  0.01643547, -0.02961645, ..., -0.00775818,\n",
       "           -0.03410022,  0.01783264],\n",
       "          [ 0.00366304,  0.03297389,  0.00117243, ..., -0.0202194 ,\n",
       "           -0.01299482,  0.00746967],\n",
       "          [ 0.01997273,  0.00774401,  0.0142099 , ..., -0.03057274,\n",
       "            0.02817659,  0.00716172]]],\n",
       " \n",
       " \n",
       "        [[[-0.03100861,  0.01528412,  0.0220536 , ...,  0.02591563,\n",
       "            0.01820394,  0.00076856],\n",
       "          [ 0.01074236, -0.00069102,  0.03431828, ..., -0.00260008,\n",
       "            0.03384195, -0.00610695],\n",
       "          [-0.01133183, -0.02966337, -0.00961806, ...,  0.00710803,\n",
       "            0.00435999, -0.01416704],\n",
       "          ...,\n",
       "          [-0.00126014,  0.03535513, -0.03022159, ...,  0.01300968,\n",
       "           -0.00042612, -0.0027754 ],\n",
       "          [ 0.02391157, -0.01935558,  0.02572431, ..., -0.00248545,\n",
       "           -0.00825652,  0.00289724],\n",
       "          [-0.00967038,  0.01250082, -0.03241204, ...,  0.01033702,\n",
       "            0.02149336,  0.02223862]],\n",
       " \n",
       "         [[ 0.00704853, -0.00088156, -0.02964266, ..., -0.01597623,\n",
       "            0.00453971, -0.0042624 ],\n",
       "          [ 0.01754694,  0.01747778, -0.00895607, ..., -0.01472466,\n",
       "            0.03421777,  0.00369744],\n",
       "          [ 0.03578421,  0.0118652 , -0.01193088, ...,  0.00100233,\n",
       "            0.00036896, -0.01892647],\n",
       "          ...,\n",
       "          [-0.00127397, -0.02927443,  0.00518175, ..., -0.02857307,\n",
       "           -0.00274492,  0.03220757],\n",
       "          [-0.00794929, -0.00285913, -0.03508723, ..., -0.02247103,\n",
       "            0.01616755, -0.01593789],\n",
       "          [ 0.02664027,  0.03181497,  0.02531314, ...,  0.02292361,\n",
       "            0.03268472,  0.02466733]],\n",
       " \n",
       "         [[-0.00348898,  0.01180875,  0.00115057, ...,  0.00464618,\n",
       "           -0.01320328,  0.02444784],\n",
       "          [ 0.03569803,  0.01884304, -0.00344473, ...,  0.02539986,\n",
       "            0.01554196, -0.0069155 ],\n",
       "          [ 0.01687291, -0.00586655, -0.00375103, ...,  0.01501999,\n",
       "            0.03533357,  0.01529415],\n",
       "          ...,\n",
       "          [-0.03574351,  0.00328474,  0.00058187, ..., -0.03097628,\n",
       "            0.01814028, -0.01138958],\n",
       "          [ 0.03316464,  0.01982843,  0.0211173 , ...,  0.02076185,\n",
       "           -0.0084323 ,  0.01860463],\n",
       "          [ 0.00080157,  0.02789797,  0.02586034, ...,  0.0260558 ,\n",
       "            0.00590381, -0.02248427]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 0.01242088,  0.04096159, -0.02569751, ..., -0.04074655,\n",
       "           -0.02823883, -0.04731227],\n",
       "          [ 0.0593933 ,  0.0462995 ,  0.06112476, ...,  0.00057274,\n",
       "            0.04174207,  0.06072143],\n",
       "          [-0.0162258 , -0.03833905,  0.02049583, ...,  0.04062611,\n",
       "           -0.01652183,  0.04643877],\n",
       "          ...,\n",
       "          [-0.04553048,  0.04303807, -0.05861805, ...,  0.03470254,\n",
       "           -0.04906276,  0.0250499 ],\n",
       "          [-0.01717626, -0.02640805, -0.04539543, ...,  0.00389293,\n",
       "            0.05521634,  0.01189475],\n",
       "          [ 0.03684746, -0.03410652,  0.0409172 , ..., -0.04840548,\n",
       "            0.05168884,  0.0315149 ]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([[[[ 0.04363248, -0.00819066,  0.02758468, ..., -0.0527084 ,\n",
       "            0.00073628,  0.02780176],\n",
       "          [-0.03707612, -0.03433565,  0.03401154, ..., -0.00225116,\n",
       "            0.05234784, -0.03950816],\n",
       "          [-0.00967206,  0.04066934,  0.0206516 , ..., -0.05617912,\n",
       "           -0.04527203,  0.04242985],\n",
       "          ...,\n",
       "          [-0.0147726 , -0.02047469, -0.04675761, ..., -0.06158943,\n",
       "           -0.05901823, -0.06094866],\n",
       "          [ 0.0484199 , -0.05989354,  0.05683939, ...,  0.05200454,\n",
       "            0.00783205,  0.06723665],\n",
       "          [-0.04355701,  0.03383642,  0.00641037, ...,  0.00310665,\n",
       "            0.01649061,  0.06135522]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.02791411,  0.05768605,  0.02605679, ...,  0.05553978,\n",
       "           -0.02716704,  0.04222023],\n",
       "          [-0.06453305,  0.02427969, -0.06638557, ...,  0.04731283,\n",
       "           -0.027686  , -0.0682486 ],\n",
       "          [-0.0577844 ,  0.00196373,  0.06459868, ...,  0.05200402,\n",
       "            0.03650883, -0.03062276],\n",
       "          ...,\n",
       "          [-0.01522402,  0.01334437,  0.05539184, ...,  0.04917491,\n",
       "            0.02123402,  0.02548587],\n",
       "          [-0.02160177, -0.0564837 ,  0.0235138 , ..., -0.03750163,\n",
       "            0.00824578,  0.0506774 ],\n",
       "          [ 0.0247567 ,  0.02823628,  0.0088363 , ...,  0.02795525,\n",
       "            0.06837177, -0.00500722]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 3.12384814e-02, -2.18842924e-03, -1.23210326e-02, ...,\n",
       "           -2.17471160e-02, -2.74603907e-02, -7.35799037e-03],\n",
       "          [-9.79199447e-03,  2.05118954e-02,  3.25629339e-02, ...,\n",
       "            2.40921900e-02,  1.64779052e-02,  2.90246680e-03],\n",
       "          [-1.60127468e-02, -2.21958570e-02,  1.42164789e-02, ...,\n",
       "           -3.45994979e-02,  1.88941509e-03, -3.03760283e-02],\n",
       "          ...,\n",
       "          [-3.67871672e-03,  2.49109901e-02, -2.34070420e-02, ...,\n",
       "            2.64095105e-02,  3.27479541e-02, -1.39338803e-02],\n",
       "          [-6.00691885e-03,  3.24881375e-02,  3.15212458e-02, ...,\n",
       "            3.67807969e-03,  2.18713805e-02, -2.83571351e-02],\n",
       "          [-1.23776495e-02, -1.68329049e-02, -1.67396385e-02, ...,\n",
       "           -3.38377096e-02, -2.22806931e-02,  2.01946795e-02]],\n",
       " \n",
       "         [[ 9.69944149e-03, -2.39390284e-02,  2.29256228e-03, ...,\n",
       "           -1.24885365e-02, -2.46788561e-03,  2.37276815e-02],\n",
       "          [-1.47021189e-02,  2.40987539e-02,  1.86461471e-02, ...,\n",
       "           -2.70410031e-02, -9.92332213e-03, -1.55549366e-02],\n",
       "          [ 2.66620740e-02, -6.07644208e-03,  3.32696885e-02, ...,\n",
       "           -1.56715345e-02,  1.33097805e-02,  2.21560262e-02],\n",
       "          ...,\n",
       "          [-1.35251507e-03,  2.86200419e-02, -2.49564666e-02, ...,\n",
       "           -2.94869505e-02,  4.06190008e-03,  1.48152411e-02],\n",
       "          [ 1.87464766e-02,  1.77763328e-02,  2.03530863e-03, ...,\n",
       "            3.11065167e-02,  1.35350004e-02,  7.46615604e-03],\n",
       "          [-2.39621885e-02, -1.93911158e-02,  1.69418380e-02, ...,\n",
       "            2.65124515e-02, -2.89549679e-03, -4.71432507e-03]],\n",
       " \n",
       "         [[ 8.49401578e-03, -2.17255224e-02,  2.47802138e-02, ...,\n",
       "           -3.15198302e-02,  9.97166336e-03,  2.14549340e-02],\n",
       "          [-1.65142082e-02,  1.11917332e-02,  1.21012293e-02, ...,\n",
       "           -6.07097894e-03,  3.22210789e-02,  3.25095132e-02],\n",
       "          [ 3.38979363e-02, -2.39774920e-02, -1.99723728e-02, ...,\n",
       "            2.30174549e-02,  1.88308321e-02,  1.45681202e-03],\n",
       "          ...,\n",
       "          [-2.42497399e-02,  2.74111703e-02,  2.14499198e-02, ...,\n",
       "            1.88646168e-02, -6.18452393e-03, -1.64698344e-02],\n",
       "          [ 2.11257935e-02, -2.31791344e-02, -2.58974843e-02, ...,\n",
       "            2.21119076e-03,  3.27858776e-02,  2.83068940e-02],\n",
       "          [ 3.16341594e-02,  2.86598578e-02,  2.80238688e-03, ...,\n",
       "            2.26913244e-02,  1.77193694e-02, -3.22122499e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 7.51024857e-03,  5.65786287e-03,  2.23213695e-02, ...,\n",
       "            1.29006729e-02,  3.16483229e-02, -8.62172991e-03],\n",
       "          [ 1.23206116e-02,  9.15744901e-03,  1.38505995e-02, ...,\n",
       "            5.68326563e-04, -1.96868330e-02,  1.32653601e-02],\n",
       "          [-2.11509243e-02,  3.53877991e-02,  3.00605819e-02, ...,\n",
       "           -1.51041895e-02,  1.86035112e-02,  3.73423100e-03],\n",
       "          ...,\n",
       "          [ 2.01206133e-02,  2.37994678e-02,  2.53106877e-02, ...,\n",
       "            3.25850248e-02, -4.70730290e-03,  3.13311815e-02],\n",
       "          [ 3.22800279e-02,  2.18326226e-02, -2.18896717e-02, ...,\n",
       "           -1.07732918e-02, -2.71831006e-02, -1.80506259e-02],\n",
       "          [ 2.60686278e-02, -2.55863834e-02, -8.67251307e-04, ...,\n",
       "           -1.13625079e-03, -2.69642100e-03, -3.26307714e-02]],\n",
       " \n",
       "         [[-3.10143605e-02,  3.09841484e-02,  3.41659114e-02, ...,\n",
       "           -2.91312765e-02,  3.30423564e-03,  2.61144638e-02],\n",
       "          [-1.82713494e-02,  3.00468281e-02, -2.44844873e-02, ...,\n",
       "            1.28252469e-02, -3.19286361e-02,  2.95411050e-03],\n",
       "          [-2.74811592e-02,  2.54382193e-02, -7.83606060e-03, ...,\n",
       "            1.19705275e-02, -1.44145656e-02, -3.22546549e-02],\n",
       "          ...,\n",
       "          [-1.91569459e-02,  3.52332070e-02,  3.27018946e-02, ...,\n",
       "            7.25326315e-03,  3.44975814e-02,  1.07482150e-02],\n",
       "          [-3.58097181e-02, -2.37304941e-02, -6.99509867e-03, ...,\n",
       "           -3.21622826e-02,  2.67333239e-02,  3.55827659e-02],\n",
       "          [-8.99858773e-04,  3.50720808e-02,  1.58985332e-03, ...,\n",
       "            1.85579397e-02, -1.63949169e-02, -2.43694969e-02]],\n",
       " \n",
       "         [[ 1.90034211e-02,  1.25050545e-02, -1.83514021e-02, ...,\n",
       "           -8.59757140e-03,  4.78301197e-04,  6.78136945e-03],\n",
       "          [-5.86980209e-03,  1.05222948e-02,  7.93695077e-03, ...,\n",
       "           -8.59085284e-03,  1.84633918e-02,  3.01175714e-02],\n",
       "          [ 2.51186267e-02, -4.34216857e-03, -1.38465669e-02, ...,\n",
       "            2.23556384e-02, -3.37867066e-03,  2.08462849e-02],\n",
       "          ...,\n",
       "          [ 3.18952650e-02, -8.45864415e-05, -2.65716817e-02, ...,\n",
       "            3.10294032e-02, -1.81069169e-02,  2.74476334e-02],\n",
       "          [-1.51640549e-03, -1.34921744e-02, -2.07801871e-02, ...,\n",
       "            2.26298198e-02, -5.77215105e-04,  1.82963908e-04],\n",
       "          [ 2.10890584e-02, -8.96832533e-03,  2.45673023e-02, ...,\n",
       "           -1.17165204e-02,  8.34264234e-03, -3.03693861e-02]]],\n",
       " \n",
       " \n",
       "        [[[-1.43791288e-02, -1.49553027e-02, -2.51317136e-02, ...,\n",
       "            6.26136735e-03, -3.32889035e-02,  4.78885323e-03],\n",
       "          [-1.26152001e-02, -3.53017598e-02,  5.33160195e-03, ...,\n",
       "           -1.40494984e-02,  2.69424543e-02, -4.03786451e-03],\n",
       "          [-1.52102318e-02,  8.94045085e-03,  1.76550336e-02, ...,\n",
       "            1.34012811e-02,  2.21048370e-02,  3.37715596e-02],\n",
       "          ...,\n",
       "          [ 2.29593553e-02,  2.79654637e-02,  2.38532983e-02, ...,\n",
       "           -1.23413783e-02,  4.02592123e-03,  3.91453877e-03],\n",
       "          [ 2.38437466e-02, -1.90783720e-02,  2.72047967e-02, ...,\n",
       "           -3.17389891e-02,  4.05552611e-03, -2.02130117e-02],\n",
       "          [ 1.25358775e-02,  6.98039681e-03,  3.15897763e-02, ...,\n",
       "           -2.10778490e-02, -1.41221862e-02, -2.44202726e-02]],\n",
       " \n",
       "         [[-2.82685142e-02,  6.13890961e-03, -2.96678320e-02, ...,\n",
       "            2.50191912e-02, -7.34716654e-03,  1.30733736e-02],\n",
       "          [-1.33349262e-02, -9.37363133e-03, -3.39670442e-02, ...,\n",
       "           -4.07594070e-03,  2.01023743e-02,  4.24553454e-03],\n",
       "          [-9.45163518e-03,  2.06029043e-03, -2.52174214e-03, ...,\n",
       "           -3.46669741e-02, -3.25782225e-03,  3.26414183e-02],\n",
       "          ...,\n",
       "          [-2.03681272e-02,  2.71214768e-02, -1.25142243e-02, ...,\n",
       "            5.97471744e-03, -1.59770623e-03,  2.58485153e-02],\n",
       "          [-3.43538001e-02, -3.48733477e-02,  2.76598781e-02, ...,\n",
       "           -2.56844163e-02,  2.08898596e-02, -2.04972029e-02],\n",
       "          [-2.37874687e-03,  4.83506173e-03, -2.51613259e-02, ...,\n",
       "           -1.67805124e-02, -7.60719739e-03,  2.17550136e-02]],\n",
       " \n",
       "         [[ 1.03542656e-02, -1.22508891e-02, -3.02719567e-02, ...,\n",
       "            2.73574889e-02,  2.37662755e-02,  1.16528645e-02],\n",
       "          [-2.10523903e-02, -2.00414658e-03,  1.10551342e-02, ...,\n",
       "            2.34867670e-02, -1.08612590e-02, -2.08832771e-02],\n",
       "          [ 1.82651468e-02,  6.23391569e-03,  2.71456465e-02, ...,\n",
       "            2.75735557e-02,  6.69477135e-03,  1.01310909e-02],\n",
       "          ...,\n",
       "          [ 3.12680081e-02, -1.09401587e-02,  2.53887326e-02, ...,\n",
       "           -2.97460426e-02, -1.51415356e-02, -7.34369084e-03],\n",
       "          [-1.08270440e-02, -1.52033847e-02, -3.38016450e-02, ...,\n",
       "           -2.63826009e-02, -3.35925557e-02, -8.56989436e-03],\n",
       "          [-4.08260897e-03,  6.37588277e-03,  2.53418796e-02, ...,\n",
       "           -7.10355863e-03,  2.91761905e-02, -1.56370699e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-0.04240749,  0.0085115 ,  0.00225066, ..., -0.03461293,\n",
       "           -0.00503231, -0.02466618],\n",
       "          [ 0.01206825, -0.06326811,  0.01714857, ...,  0.05959077,\n",
       "           -0.028015  ,  0.01849967],\n",
       "          [-0.0626149 , -0.03200582, -0.03559746, ..., -0.0135496 ,\n",
       "            0.01296214, -0.05108225],\n",
       "          ...,\n",
       "          [ 0.06170325, -0.04318825,  0.03421206, ..., -0.03900655,\n",
       "           -0.00030035, -0.05924315],\n",
       "          [-0.01324837, -0.01463357, -0.057868  , ...,  0.05066717,\n",
       "            0.05959135, -0.02020209],\n",
       "          [ 0.03931784,  0.06730023,  0.06712134, ..., -0.05684084,\n",
       "            0.03741687, -0.01756121]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.02851893, -0.02886669,  0.06602526, ..., -0.03220872,\n",
       "            0.0039215 , -0.009687  ],\n",
       "          [ 0.03513381, -0.00424433,  0.05747625, ..., -0.05186822,\n",
       "            0.05964689, -0.01491283],\n",
       "          [ 0.03691976, -0.04472353, -0.06647132, ..., -0.04673247,\n",
       "            0.00362539, -0.05053447],\n",
       "          ...,\n",
       "          [ 0.02515242, -0.05165156, -0.05733957, ..., -0.0577248 ,\n",
       "            0.01188848,  0.02919558],\n",
       "          [ 0.00201482, -0.0273318 , -0.03782944, ...,  0.03913323,\n",
       "            0.04745341, -0.02383123],\n",
       "          [ 0.06074616,  0.03839036, -0.01532771, ..., -0.0604051 ,\n",
       "            0.04571401, -0.01351247]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 0.00264508,  0.02034114,  0.03322123, ..., -0.02552324,\n",
       "           -0.01359772,  0.03363896],\n",
       "          [-0.03489861, -0.00855749,  0.00455936, ...,  0.03567492,\n",
       "           -0.02461163, -0.01389363],\n",
       "          [ 0.02379388,  0.01070528,  0.01183434, ...,  0.03476437,\n",
       "            0.00775301,  0.01842613],\n",
       "          ...,\n",
       "          [-0.00421754, -0.00119816, -0.00317489, ...,  0.00950179,\n",
       "           -0.03200525, -0.01645999],\n",
       "          [-0.01975064,  0.03071415,  0.02621725, ..., -0.01433347,\n",
       "            0.01766638,  0.03256469],\n",
       "          [-0.01693531, -0.01448707, -0.0140253 , ...,  0.00114075,\n",
       "            0.02920559,  0.01470142]],\n",
       " \n",
       "         [[-0.02197069,  0.01929328,  0.00601859, ..., -0.01561258,\n",
       "           -0.01494839, -0.01762013],\n",
       "          [ 0.01905654,  0.02932961,  0.00981339, ..., -0.01158171,\n",
       "           -0.01794338,  0.01005744],\n",
       "          [-0.00133424, -0.01025639, -0.03337965, ...,  0.03124538,\n",
       "           -0.01050452, -0.02880373],\n",
       "          ...,\n",
       "          [ 0.03086594, -0.00636102,  0.01030441, ...,  0.00047898,\n",
       "           -0.0279978 , -0.02628201],\n",
       "          [-0.02745035, -0.02513064,  0.01808261, ...,  0.00031092,\n",
       "            0.03089313,  0.01881972],\n",
       "          [-0.01981629,  0.00046588, -0.01876497, ...,  0.02346658,\n",
       "           -0.03412298,  0.03406421]],\n",
       " \n",
       "         [[-0.03011698, -0.00623989,  0.02824566, ...,  0.01611616,\n",
       "            0.01043575,  0.01228061],\n",
       "          [-0.02255887,  0.03148095, -0.02195629, ..., -0.03275165,\n",
       "           -0.02763097, -0.03321801],\n",
       "          [ 0.00117027,  0.03145278, -0.00872698, ...,  0.0156056 ,\n",
       "           -0.01020071, -0.00802998],\n",
       "          ...,\n",
       "          [ 0.01884047, -0.00627445,  0.00337339, ..., -0.00019756,\n",
       "           -0.02254702, -0.01541991],\n",
       "          [ 0.00600626,  0.01545108, -0.02457462, ...,  0.01427715,\n",
       "           -0.03464288, -0.00348515],\n",
       "          [-0.00048038,  0.02531175, -0.03337194, ...,  0.02475815,\n",
       "            0.00107437,  0.02780255]]],\n",
       " \n",
       " \n",
       "        [[[ 0.03194216, -0.00441659,  0.00333589, ..., -0.00197568,\n",
       "           -0.02624821, -0.02327444],\n",
       "          [ 0.02595776,  0.0335445 , -0.01742345, ..., -0.02800336,\n",
       "           -0.02608036, -0.01439874],\n",
       "          [-0.01402734,  0.02208531, -0.01447879, ...,  0.02922843,\n",
       "            0.00724056, -0.02599099],\n",
       "          ...,\n",
       "          [ 0.01767283, -0.01179045,  0.03403131, ..., -0.02865987,\n",
       "            0.03071807,  0.01822864],\n",
       "          [ 0.00632229,  0.0301237 , -0.010627  , ..., -0.02073636,\n",
       "            0.03312513, -0.03507992],\n",
       "          [-0.00080498,  0.02059056, -0.01555519, ...,  0.00301914,\n",
       "           -0.02984473, -0.00458617]],\n",
       " \n",
       "         [[ 0.00859498, -0.0045944 ,  0.01859758, ...,  0.00375825,\n",
       "            0.01362116, -0.0138828 ],\n",
       "          [-0.01255747, -0.02162971, -0.02444814, ...,  0.02611571,\n",
       "            0.03154906,  0.00765385],\n",
       "          [-0.00535387, -0.01561753, -0.02398549, ...,  0.0302721 ,\n",
       "           -0.03458193, -0.01824194],\n",
       "          ...,\n",
       "          [ 0.02242486,  0.0153136 ,  0.0225447 , ...,  0.02267893,\n",
       "           -0.00904432,  0.02966926],\n",
       "          [-0.01328345,  0.01761838,  0.02935668, ...,  0.03351858,\n",
       "           -0.01813983, -0.02442635],\n",
       "          [-0.01963208, -0.00436233,  0.02155713, ..., -0.0001605 ,\n",
       "            0.02512466, -0.0166566 ]],\n",
       " \n",
       "         [[ 0.0094978 ,  0.01658778, -0.02420655, ..., -0.03224739,\n",
       "            0.01441866,  0.03078207],\n",
       "          [ 0.0293415 , -0.01753406, -0.03003435, ..., -0.00514035,\n",
       "           -0.03525477,  0.02352881],\n",
       "          [ 0.01609324, -0.03315284, -0.02825693, ..., -0.0142764 ,\n",
       "           -0.03551723, -0.00772719],\n",
       "          ...,\n",
       "          [-0.01604745,  0.0308641 ,  0.0052801 , ..., -0.01910073,\n",
       "            0.00368288, -0.00930593],\n",
       "          [ 0.00325219, -0.01086315,  0.01133978, ..., -0.00693872,\n",
       "           -0.03309338, -0.00075064],\n",
       "          [ 0.02032368,  0.03507826,  0.00313659, ...,  0.0026917 ,\n",
       "           -0.01901578,  0.02273443]]],\n",
       " \n",
       " \n",
       "        [[[-0.02389374, -0.02548665,  0.03389227, ..., -0.01732903,\n",
       "            0.02545686,  0.01857305],\n",
       "          [ 0.01118065,  0.03349564,  0.02819429, ...,  0.03362101,\n",
       "            0.00873716,  0.00333877],\n",
       "          [-0.00747143, -0.0159298 , -0.03475674, ..., -0.01198118,\n",
       "           -0.01051642,  0.00761255],\n",
       "          ...,\n",
       "          [ 0.00418548,  0.02206773, -0.03342994, ...,  0.02221758,\n",
       "           -0.01425096, -0.00988523],\n",
       "          [-0.00565914,  0.00968895,  0.031516  , ...,  0.0200313 ,\n",
       "           -0.01398587,  0.00419461],\n",
       "          [-0.00704606,  0.00735034, -0.03379599, ..., -0.00263542,\n",
       "            0.0222173 , -0.03058299]],\n",
       " \n",
       "         [[-0.01517671, -0.01590703,  0.03595831, ..., -0.00479364,\n",
       "           -0.02956553, -0.0263618 ],\n",
       "          [-0.01371678,  0.01336448, -0.00883865, ...,  0.02935456,\n",
       "           -0.00529558, -0.00554914],\n",
       "          [ 0.02106246,  0.02753614,  0.00832011, ...,  0.01602226,\n",
       "            0.02902228,  0.0150636 ],\n",
       "          ...,\n",
       "          [ 0.00091588, -0.01755006, -0.02719523, ...,  0.0289358 ,\n",
       "            0.03367592, -0.02001228],\n",
       "          [-0.00949366,  0.00193916,  0.00929972, ..., -0.01512722,\n",
       "           -0.01492192,  0.024994  ],\n",
       "          [ 0.01457971, -0.00940695, -0.01690696, ..., -0.02669522,\n",
       "            0.01881726, -0.0056816 ]],\n",
       " \n",
       "         [[-0.0124518 ,  0.01849629,  0.0032081 , ..., -0.02036024,\n",
       "           -0.01705376, -0.00015258],\n",
       "          [-0.0166004 , -0.02240147,  0.0285567 , ...,  0.01945559,\n",
       "            0.02554357,  0.00711819],\n",
       "          [-0.02285699, -0.03447791,  0.0002839 , ..., -0.00345032,\n",
       "            0.00460435,  0.03186945],\n",
       "          ...,\n",
       "          [-0.03063136,  0.00276786, -0.00813885, ...,  0.02239536,\n",
       "            0.01406913,  0.02723072],\n",
       "          [-0.03019756,  0.00802788, -0.02054461, ..., -0.02476745,\n",
       "           -0.00145502, -0.03545418],\n",
       "          [ 0.00833472, -0.01943843, -0.03277571, ..., -0.02240694,\n",
       "            0.03374846,  0.00806046]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 0.05958672,  0.02228151,  0.00122367, ...,  0.00472781,\n",
       "            0.04714673, -0.04054859],\n",
       "          [-0.01818267,  0.02400479, -0.00362992, ...,  0.00686869,\n",
       "           -0.0011759 ,  0.02210306],\n",
       "          [-0.02232298,  0.03563618,  0.05564337, ..., -0.06249355,\n",
       "            0.05768403,  0.00242595],\n",
       "          ...,\n",
       "          [ 0.00389359, -0.02378016, -0.05609579, ...,  0.00268615,\n",
       "           -0.06019174,  0.03731728],\n",
       "          [ 0.06704924, -0.00946725,  0.04475483, ...,  0.05565707,\n",
       "            0.03564096,  0.04921539],\n",
       "          [ 0.05036033, -0.05319301,  0.0601389 , ..., -0.01129368,\n",
       "            0.03540932, -0.04374904]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.06070766,  0.04056074,  0.00839376, ...,  0.01234958,\n",
       "            0.03409352,  0.00860044],\n",
       "          [ 0.0424232 , -0.06352111, -0.04093716, ..., -0.03641234,\n",
       "           -0.03382941,  0.0420589 ],\n",
       "          [ 0.00150832,  0.04182079,  0.05899744, ...,  0.04074175,\n",
       "           -0.00289021, -0.04439718],\n",
       "          ...,\n",
       "          [-0.01887872,  0.01309336, -0.05424204, ...,  0.01259702,\n",
       "            0.00323303,  0.00687338],\n",
       "          [-0.05361848, -0.05091814,  0.04468883, ..., -0.03315347,\n",
       "           -0.04896998, -0.03162937],\n",
       "          [ 0.05672194,  0.05172079, -0.05093855, ...,  0.05741882,\n",
       "            0.03508878,  0.04396433]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-2.52446830e-02, -2.48465277e-02, -3.41398716e-02, ...,\n",
       "           -2.31458582e-02, -1.32767856e-02, -4.91243042e-03],\n",
       "          [-7.42364116e-03, -3.00291218e-02,  1.85842812e-03, ...,\n",
       "           -6.34718686e-04, -4.45241854e-03,  1.95898749e-02],\n",
       "          [ 1.18692182e-02,  2.70942375e-02,  1.96445659e-02, ...,\n",
       "           -2.80232728e-03,  1.26460418e-02,  1.78127997e-02],\n",
       "          ...,\n",
       "          [-2.14327909e-02,  2.26632357e-02, -2.79991217e-02, ...,\n",
       "            3.03604826e-02, -1.75784063e-02, -2.03720108e-03],\n",
       "          [-1.24035794e-02,  1.57803409e-02,  4.14459407e-03, ...,\n",
       "            1.58221275e-03, -2.35120859e-02,  3.47109288e-02],\n",
       "          [ 8.28268752e-03, -3.50780413e-02, -2.41692327e-02, ...,\n",
       "            1.94270350e-02, -3.48193794e-02,  2.97593251e-02]],\n",
       " \n",
       "         [[ 3.53640169e-02, -2.79435143e-03, -3.36942449e-02, ...,\n",
       "            6.76992536e-03,  8.83322209e-04,  1.37760714e-02],\n",
       "          [-2.05686092e-02,  1.03060976e-02, -4.23645973e-03, ...,\n",
       "            8.51929188e-04, -3.23552713e-02,  1.00659467e-02],\n",
       "          [-2.47038864e-02,  7.46023655e-03, -2.91505121e-02, ...,\n",
       "           -1.42485760e-02,  2.12166682e-02, -2.82399096e-02],\n",
       "          ...,\n",
       "          [ 3.24769467e-02,  2.88353711e-02,  3.58367190e-02, ...,\n",
       "            2.71273032e-02,  1.85756534e-02, -3.34256962e-02],\n",
       "          [ 2.24934593e-02,  3.50182056e-02, -1.44045055e-03, ...,\n",
       "           -1.93902645e-02, -3.55054438e-03,  2.17540488e-02],\n",
       "          [ 2.81741023e-02,  3.02777365e-02, -1.48489662e-02, ...,\n",
       "            1.75973400e-02, -3.10141016e-02, -3.34311500e-02]],\n",
       " \n",
       "         [[ 3.48008573e-02, -1.84388198e-02, -1.06543098e-02, ...,\n",
       "            1.17604397e-02,  1.47519559e-02, -3.05878129e-02],\n",
       "          [ 3.60805541e-03,  1.62155405e-02, -1.94863975e-03, ...,\n",
       "            2.19469778e-02, -1.98580194e-02,  2.27374211e-02],\n",
       "          [-9.56685841e-03,  4.37214971e-03,  2.81373262e-02, ...,\n",
       "            3.09537873e-02, -3.14897038e-02,  1.90793872e-02],\n",
       "          ...,\n",
       "          [-1.67529993e-02,  1.22248642e-02,  2.16687396e-02, ...,\n",
       "           -2.61691380e-02, -1.97681841e-02, -1.31267197e-02],\n",
       "          [-2.18604114e-02, -1.44693628e-03,  4.19553369e-03, ...,\n",
       "            9.83354077e-03,  3.37460190e-02, -1.96412280e-02],\n",
       "          [ 7.30723888e-03, -2.37719640e-02, -5.62570989e-04, ...,\n",
       "            3.08382660e-02,  1.97598487e-02,  5.96477091e-03]]],\n",
       " \n",
       " \n",
       "        [[[ 2.32685469e-02, -2.82888170e-02, -2.34902501e-02, ...,\n",
       "           -9.73081775e-03,  1.13084540e-03,  1.38535090e-02],\n",
       "          [-2.40603164e-02,  2.36778855e-02, -7.90220127e-03, ...,\n",
       "           -1.65648386e-02, -2.83309557e-02,  2.05221325e-02],\n",
       "          [ 2.34401114e-02,  3.55065912e-02, -2.21432243e-02, ...,\n",
       "            6.83812425e-03, -1.76270753e-02,  1.60092860e-02],\n",
       "          ...,\n",
       "          [-7.74417818e-03, -9.62313265e-03,  6.41378015e-03, ...,\n",
       "            2.89123654e-02, -7.27301650e-03,  3.61610949e-03],\n",
       "          [-7.09000044e-03,  2.00995468e-02, -2.35788301e-02, ...,\n",
       "            2.46154889e-02, -3.09515558e-02,  3.14961746e-03],\n",
       "          [-1.90342385e-02,  3.72756273e-03, -4.32576984e-04, ...,\n",
       "           -1.55010633e-02,  2.14913674e-02,  1.94249600e-02]],\n",
       " \n",
       "         [[-1.78631973e-02, -3.31292488e-02, -2.73673814e-02, ...,\n",
       "           -3.15971486e-02, -1.04882196e-03, -2.79924124e-02],\n",
       "          [-1.01370439e-02, -2.98774485e-02,  1.29086226e-02, ...,\n",
       "            3.42827439e-02,  2.68655419e-02,  3.32690552e-02],\n",
       "          [-1.73615199e-02,  7.72733241e-03, -1.48646329e-02, ...,\n",
       "            2.66995132e-02, -2.50732824e-02, -1.73715595e-02],\n",
       "          ...,\n",
       "          [ 2.31040008e-02, -4.95122932e-03,  2.87988633e-02, ...,\n",
       "            6.29037619e-03, -3.16090882e-02, -2.34540664e-02],\n",
       "          [-9.92882065e-03, -7.42170960e-04,  3.03222984e-02, ...,\n",
       "            2.48669684e-02,  1.18216351e-02, -3.81734595e-03],\n",
       "          [-1.42175537e-02, -1.27129927e-02, -1.40057001e-02, ...,\n",
       "            2.60640755e-02,  1.01758428e-02,  1.99048966e-02]],\n",
       " \n",
       "         [[-1.38823651e-02, -2.71121506e-02, -1.12337004e-02, ...,\n",
       "            1.88307911e-02,  7.98561051e-03,  7.23945349e-03],\n",
       "          [ 2.94304490e-02, -8.98512825e-03,  6.50609285e-03, ...,\n",
       "           -2.71513816e-02, -2.84700785e-02,  3.57808173e-02],\n",
       "          [-4.60579619e-03,  9.95462015e-03,  2.86154747e-02, ...,\n",
       "           -3.20129842e-02,  2.57503279e-02,  1.03302635e-02],\n",
       "          ...,\n",
       "          [ 7.27330148e-03, -3.09018735e-02, -1.07180327e-03, ...,\n",
       "            2.46018693e-02,  2.69982293e-02,  3.64603847e-03],\n",
       "          [-2.96099056e-02, -3.12071927e-02,  2.87066922e-02, ...,\n",
       "            3.24838012e-02,  1.82105005e-02,  2.47312598e-02],\n",
       "          [ 2.41572037e-02,  3.59127820e-02, -1.09210610e-03, ...,\n",
       "           -1.06357094e-02,  1.08784847e-02, -1.55481044e-02]]],\n",
       " \n",
       " \n",
       "        [[[-1.88494585e-02,  2.14566737e-02, -1.64132677e-02, ...,\n",
       "           -1.97984502e-02, -3.56618986e-02, -1.50311310e-02],\n",
       "          [-3.52976210e-02, -4.25706431e-03,  8.65050778e-03, ...,\n",
       "            2.33925208e-02,  2.01684125e-02,  1.46986507e-02],\n",
       "          [-1.10601820e-02,  1.15031525e-02,  1.36078261e-02, ...,\n",
       "           -1.46056004e-02, -2.80254297e-02, -2.46365145e-02],\n",
       "          ...,\n",
       "          [-1.91546232e-02, -1.32686123e-02,  2.99657881e-02, ...,\n",
       "           -3.60461511e-02,  2.76224241e-02, -7.34932534e-03],\n",
       "          [ 1.77371539e-02,  1.87246017e-02,  2.70535201e-02, ...,\n",
       "            3.00648957e-02,  2.76157111e-02,  6.74097613e-03],\n",
       "          [ 2.52640396e-02,  1.52768195e-02, -1.16289314e-02, ...,\n",
       "           -6.59264624e-05, -8.53295252e-03, -7.81848468e-03]],\n",
       " \n",
       "         [[-1.42764598e-02,  1.53158121e-02,  2.97402516e-02, ...,\n",
       "            1.49202868e-02, -3.25935334e-03,  3.08575854e-02],\n",
       "          [-3.56591456e-02, -1.99012328e-02,  3.46042141e-02, ...,\n",
       "           -2.09088288e-02, -1.30480621e-02,  3.13155353e-03],\n",
       "          [ 2.00359412e-02,  1.33838430e-02, -9.42801125e-03, ...,\n",
       "           -1.47167537e-02, -2.05269456e-03,  7.01099634e-06],\n",
       "          ...,\n",
       "          [ 2.11318582e-02, -9.49735381e-03,  1.32694133e-02, ...,\n",
       "            1.99388452e-02, -2.85376310e-02,  3.19069996e-02],\n",
       "          [-2.10268740e-02, -8.11622292e-03, -1.42173842e-03, ...,\n",
       "           -2.78587006e-02,  1.69595368e-02, -1.04954951e-02],\n",
       "          [ 2.16861963e-02, -3.53966281e-02,  2.91032419e-02, ...,\n",
       "           -2.80525833e-02,  2.84070149e-02,  4.64764982e-03]],\n",
       " \n",
       "         [[-1.56172495e-02,  1.05597079e-02, -5.88359125e-03, ...,\n",
       "           -3.19534503e-02,  5.92321157e-03, -2.00778730e-02],\n",
       "          [ 2.54257210e-02, -3.49335261e-02,  5.57651743e-03, ...,\n",
       "           -2.24012081e-02,  6.46650791e-04, -9.48277116e-03],\n",
       "          [ 2.75789574e-02,  2.02076957e-02, -8.63711163e-03, ...,\n",
       "           -3.22200283e-02,  1.14283562e-02,  3.31165269e-03],\n",
       "          ...,\n",
       "          [-3.00608855e-02,  2.01768018e-02,  2.98848525e-02, ...,\n",
       "            2.58168206e-02, -2.94450093e-02, -1.69769134e-02],\n",
       "          [ 2.65472382e-02, -3.59799303e-02, -9.37084295e-03, ...,\n",
       "            6.83205202e-03, -2.24684663e-02, -2.68092677e-02],\n",
       "          [-8.01452436e-03, -1.53843183e-02, -2.28978321e-03, ...,\n",
       "            3.01512703e-03, -3.39923054e-03,  8.67693499e-03]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-1.5098609e-02,  6.5430403e-03,  4.6549894e-02, ...,\n",
       "           -1.6738165e-02, -4.0389672e-02,  3.8474768e-02],\n",
       "          [ 3.4699515e-03, -1.1964582e-02,  4.3214105e-02, ...,\n",
       "           -2.6317947e-02,  6.8187416e-03, -3.6151707e-02],\n",
       "          [ 8.0950558e-05,  4.5707539e-02, -2.1533329e-02, ...,\n",
       "            1.5104845e-02, -9.5191747e-03, -1.2184296e-02],\n",
       "          ...,\n",
       "          [ 6.3515157e-03,  5.0243653e-02,  3.4157149e-02, ...,\n",
       "            4.1360110e-03,  6.6921711e-03, -4.4341959e-02],\n",
       "          [ 4.0901750e-02,  3.0678645e-02,  5.2002840e-02, ...,\n",
       "            3.5947174e-02, -4.3823235e-02,  4.2567737e-02],\n",
       "          [ 1.1488460e-02,  2.9479578e-02,  5.8527619e-02, ...,\n",
       "           -2.6794411e-02,  6.1732903e-02, -4.9660351e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[ 0.05113316, -0.037183  , -0.00606594, ..., -0.03644319,\n",
       "           -0.03914421, -0.05050806],\n",
       "          [-0.05288029,  0.05158833, -0.03243206, ...,  0.01077644,\n",
       "            0.06405422,  0.06661008],\n",
       "          [-0.05658887, -0.01720817, -0.00392091, ..., -0.00608548,\n",
       "            0.04114909, -0.02543304],\n",
       "          ...,\n",
       "          [ 0.06610806,  0.01666747,  0.00461043, ...,  0.0057579 ,\n",
       "            0.01433562, -0.06091111],\n",
       "          [-0.0664177 ,  0.03237455, -0.02231915, ...,  0.04594247,\n",
       "           -0.00474708,  0.03652074],\n",
       "          [ 0.06624895,  0.01208095,  0.00795379, ..., -0.05173618,\n",
       "            0.05767624, -0.00989141]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-2.59289891e-02, -1.68818142e-02,  2.32544802e-02, ...,\n",
       "           -9.30933841e-03, -1.10141300e-02, -2.10641697e-02],\n",
       "          [-3.18159014e-02,  1.12467520e-02, -2.88488679e-02, ...,\n",
       "           -3.25737670e-02, -1.20500308e-02, -2.26665810e-02],\n",
       "          [-6.31678849e-04,  1.57819130e-02,  2.53564641e-02, ...,\n",
       "            1.69320479e-02,  1.79120973e-02,  1.26499906e-02],\n",
       "          ...,\n",
       "          [-7.16438331e-03,  2.82488391e-02,  2.63272971e-03, ...,\n",
       "           -4.32444364e-03, -2.84631774e-02, -1.26527622e-02],\n",
       "          [ 3.42073888e-02, -1.90598238e-02,  2.64773443e-02, ...,\n",
       "           -2.88087595e-02, -2.96567678e-02, -1.05927121e-02],\n",
       "          [ 1.84415989e-02,  2.98321918e-02, -2.97950730e-02, ...,\n",
       "            3.42415944e-02, -2.26587281e-02,  7.66800717e-03]],\n",
       " \n",
       "         [[ 1.75547376e-02,  3.04342285e-02,  7.61893392e-03, ...,\n",
       "           -1.45761427e-02,  2.20263936e-02,  4.64051589e-03],\n",
       "          [-3.52155566e-02, -1.53988488e-02,  2.36317553e-02, ...,\n",
       "           -9.14980099e-03,  2.63631828e-02,  3.44126448e-02],\n",
       "          [-2.68863440e-02,  6.59167022e-03,  2.81008780e-02, ...,\n",
       "           -1.35123059e-02, -2.05096155e-02, -2.92464979e-02],\n",
       "          ...,\n",
       "          [-2.58304998e-02, -1.35093555e-02,  8.81802663e-03, ...,\n",
       "            1.82593837e-02,  3.25270742e-02, -7.16692954e-03],\n",
       "          [ 5.00265136e-03, -7.81600550e-03,  1.35792233e-02, ...,\n",
       "            1.21903680e-02,  3.48445624e-02, -1.71143152e-02],\n",
       "          [ 1.13488361e-02, -1.62693113e-03, -1.29888803e-02, ...,\n",
       "           -1.00911371e-02, -2.10762136e-02, -2.23194342e-02]],\n",
       " \n",
       "         [[ 1.09433532e-02,  1.87855810e-02, -3.10786255e-02, ...,\n",
       "            3.06064412e-02,  2.87994742e-02,  2.28989273e-02],\n",
       "          [-3.42592001e-02,  2.88311094e-02, -2.43564378e-02, ...,\n",
       "           -3.19709033e-02,  1.31330788e-02, -1.14445649e-02],\n",
       "          [-9.31730494e-03, -3.16837728e-02,  1.06800869e-02, ...,\n",
       "           -4.15738672e-03, -1.94246322e-03,  3.71029601e-03],\n",
       "          ...,\n",
       "          [ 1.92315094e-02, -2.79599689e-02,  1.44113824e-02, ...,\n",
       "            4.33082879e-03, -1.57135110e-02,  1.67362057e-02],\n",
       "          [-1.73697453e-02,  3.20981964e-02,  2.72055790e-02, ...,\n",
       "            1.86784975e-02, -2.26320401e-02, -9.61150043e-03],\n",
       "          [-1.38358474e-02, -3.18637192e-02, -2.45674327e-03, ...,\n",
       "            2.55697966e-03, -3.51860002e-03, -9.87252034e-03]]],\n",
       " \n",
       " \n",
       "        [[[ 1.05190501e-02,  2.15356164e-02, -1.34387761e-02, ...,\n",
       "            2.65182629e-02, -3.40238400e-02, -5.85856475e-03],\n",
       "          [ 3.34654525e-02,  5.21908328e-03, -2.49150507e-02, ...,\n",
       "           -1.99720711e-02, -7.43556395e-03,  1.44943856e-02],\n",
       "          [-2.83145849e-02, -3.56945023e-02, -1.50802210e-02, ...,\n",
       "            3.19607332e-02, -3.35783958e-02,  1.57545730e-02],\n",
       "          ...,\n",
       "          [ 2.27164477e-03, -3.52792181e-02,  1.41148828e-02, ...,\n",
       "            2.46401019e-02, -8.79346579e-03,  7.80310482e-04],\n",
       "          [-1.48107782e-02, -2.17227265e-02,  3.05437297e-03, ...,\n",
       "            1.06059350e-02,  3.41470912e-03, -3.50469314e-02],\n",
       "          [-2.97464728e-02, -8.40481371e-04,  3.48734036e-02, ...,\n",
       "            2.99059451e-02,  2.40630955e-02,  1.61703900e-02]],\n",
       " \n",
       "         [[-2.44305972e-02, -3.40342596e-02, -3.40326428e-02, ...,\n",
       "           -2.45582424e-02, -2.09578089e-02, -1.12284273e-02],\n",
       "          [ 2.31579095e-02,  2.57728659e-02, -2.39414535e-02, ...,\n",
       "            2.26463377e-03,  1.80148035e-02, -2.29151957e-02],\n",
       "          [-2.04020943e-02,  9.67767462e-03,  6.56942278e-03, ...,\n",
       "           -1.14298798e-02,  2.39309072e-02, -2.58804671e-02],\n",
       "          ...,\n",
       "          [ 2.61536352e-02, -3.51559781e-02, -1.20492317e-02, ...,\n",
       "            1.30880140e-02,  3.57509404e-02, -2.58888453e-02],\n",
       "          [ 1.89737231e-02,  1.54368244e-02,  3.46608758e-02, ...,\n",
       "           -2.04128902e-02, -3.60219069e-02,  3.11492682e-02],\n",
       "          [ 1.13796517e-02,  3.14577520e-02, -1.75901577e-02, ...,\n",
       "           -1.18771680e-02, -2.98737995e-02, -2.16417182e-02]],\n",
       " \n",
       "         [[ 3.12911049e-02,  9.66738909e-04,  1.32057816e-04, ...,\n",
       "           -1.19932164e-02,  1.22101605e-03, -2.23229267e-02],\n",
       "          [-3.38109024e-02,  6.72367588e-03, -1.32292621e-02, ...,\n",
       "           -1.82711091e-02,  2.50294134e-02,  3.45855504e-02],\n",
       "          [ 2.16427930e-02, -1.40804611e-02, -1.60131976e-03, ...,\n",
       "           -3.31334770e-04, -2.70107966e-02,  1.79018080e-02],\n",
       "          ...,\n",
       "          [-9.83806513e-03, -2.58810688e-02,  1.72940604e-02, ...,\n",
       "            2.35969909e-02,  1.29265226e-02, -1.09486934e-02],\n",
       "          [ 3.82476300e-03,  2.59366892e-02,  2.05242410e-02, ...,\n",
       "           -3.01670134e-02,  3.12291458e-03,  1.93322450e-03],\n",
       "          [-9.62948985e-03,  7.25303218e-03,  9.47043300e-03, ...,\n",
       "           -6.75616227e-03,  7.91586936e-04, -1.26177818e-02]]],\n",
       " \n",
       " \n",
       "        [[[-8.80154409e-03, -1.24815498e-02,  3.14770937e-02, ...,\n",
       "           -3.45138013e-02, -1.87052097e-02, -1.36048850e-02],\n",
       "          [-1.47444978e-02, -4.42086160e-03, -2.40459926e-02, ...,\n",
       "           -5.77450357e-03, -2.66086161e-02,  3.25865000e-02],\n",
       "          [-1.91619787e-02, -2.20035855e-02, -3.43751535e-03, ...,\n",
       "            1.54872797e-02, -3.31975743e-02, -2.23162510e-02],\n",
       "          ...,\n",
       "          [ 5.80393523e-03,  1.10311992e-02, -4.07826528e-03, ...,\n",
       "           -3.43858190e-02,  1.23193786e-02, -2.80604884e-02],\n",
       "          [-2.70392988e-02, -2.39997394e-02, -4.80764359e-03, ...,\n",
       "            1.76038556e-02, -2.53699459e-02,  2.14689225e-03],\n",
       "          [ 2.52390988e-02, -3.90402973e-04,  8.00168142e-03, ...,\n",
       "           -5.45923039e-03, -1.44398939e-02,  2.69225389e-02]],\n",
       " \n",
       "         [[-2.56242603e-03,  2.35912614e-02,  1.94128826e-02, ...,\n",
       "           -3.21975350e-03,  3.42099294e-02,  1.29384734e-02],\n",
       "          [ 1.71113014e-03, -1.48311332e-02, -1.96349733e-02, ...,\n",
       "            2.82467604e-02, -1.26966555e-02, -2.48257760e-02],\n",
       "          [-3.70598584e-03, -2.81367898e-02,  2.81718224e-02, ...,\n",
       "           -3.57439965e-02, -1.82760898e-02,  3.55278254e-02],\n",
       "          ...,\n",
       "          [ 3.34239751e-03, -6.07283600e-03, -3.38759273e-03, ...,\n",
       "            1.28300153e-02,  1.49017125e-02, -2.75823660e-02],\n",
       "          [ 1.09879002e-02, -2.71959119e-02,  5.67109510e-03, ...,\n",
       "            1.85711719e-02, -2.13087574e-02, -2.88654044e-02],\n",
       "          [ 2.87079364e-02,  8.36934522e-03,  1.63544491e-02, ...,\n",
       "           -1.34704262e-03, -1.70206092e-02, -1.19195990e-02]],\n",
       " \n",
       "         [[-8.77240486e-03, -2.88934503e-02, -1.45833939e-04, ...,\n",
       "            1.43879130e-02,  4.14297730e-03, -3.49167660e-02],\n",
       "          [-1.37613174e-02, -2.38306820e-05, -2.84672901e-02, ...,\n",
       "            1.17577128e-02,  2.41897628e-03, -3.09885330e-02],\n",
       "          [-5.84046356e-03,  3.37767005e-02, -1.57433208e-02, ...,\n",
       "           -1.57393720e-02, -4.48793173e-03, -3.64372507e-03],\n",
       "          ...,\n",
       "          [-1.93739701e-02, -1.46380253e-02, -2.36118734e-02, ...,\n",
       "            2.44055316e-03, -8.22856277e-03,  1.86867975e-02],\n",
       "          [ 2.23241486e-02, -1.93217397e-03, -2.39226483e-02, ...,\n",
       "            1.06138065e-02,  1.22302771e-02,  1.02129579e-02],\n",
       "          [ 2.94347703e-02, -2.88825668e-02, -2.93213557e-02, ...,\n",
       "           -1.04154684e-02, -3.34004201e-02,  9.47670639e-03]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 0.0300717 , -0.04730727,  0.00269823, ...,  0.06753869,\n",
       "            0.06450251,  0.03810757],\n",
       "          [ 0.04164314,  0.04439215, -0.00334736, ..., -0.05413987,\n",
       "            0.03481802, -0.01915011],\n",
       "          [-0.04143877,  0.03343345,  0.00024542, ...,  0.05933399,\n",
       "            0.02904479, -0.06423014],\n",
       "          ...,\n",
       "          [ 0.00383561,  0.05261074, -0.03798234, ...,  0.04397315,\n",
       "            0.02498318, -0.0169468 ],\n",
       "          [ 0.0111798 ,  0.01160575, -0.00429482, ..., -0.05787528,\n",
       "            0.06542549,  0.05503955],\n",
       "          [-0.00759855, -0.00572948,  0.02987859, ..., -0.00042251,\n",
       "            0.04871857,  0.0217021 ]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.00337217, -0.06111579,  0.04416811, ..., -0.03956078,\n",
       "            0.05506108, -0.06107991],\n",
       "          [ 0.05420586, -0.0090501 ,  0.00891831, ..., -0.02841125,\n",
       "           -0.04247583, -0.00475591],\n",
       "          [ 0.06633747,  0.01722364, -0.06248448, ...,  0.01185793,\n",
       "           -0.06616958,  0.06747478],\n",
       "          ...,\n",
       "          [ 0.03607668,  0.04035046, -0.06345193, ...,  0.03462602,\n",
       "           -0.05896296, -0.04949775],\n",
       "          [-0.00222782,  0.05688627,  0.0514788 , ...,  0.03304009,\n",
       "           -0.022788  , -0.04489429],\n",
       "          [-0.04439269,  0.0046708 , -0.00322477, ...,  0.04534358,\n",
       "            0.01710168, -0.04138094]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[ 1.15999132e-02,  9.64263454e-03,  3.02780494e-02, ...,\n",
       "           -1.92058794e-02,  2.44422629e-02,  2.12526545e-02],\n",
       "          [ 2.93422714e-02, -2.44839974e-02,  1.86949037e-02, ...,\n",
       "            1.28429718e-02, -1.23682115e-02,  3.07623297e-04],\n",
       "          [-3.32330614e-02, -5.52941673e-03, -2.51408406e-02, ...,\n",
       "           -1.23289544e-02,  8.01824033e-04, -3.56600471e-02],\n",
       "          ...,\n",
       "          [ 1.47408508e-02,  1.55838951e-02, -1.51650235e-03, ...,\n",
       "           -2.69630477e-02, -3.35567072e-02, -1.26279071e-02],\n",
       "          [-5.50761633e-03,  1.54714175e-02,  5.09489328e-04, ...,\n",
       "            2.83541381e-02,  3.10830846e-02, -7.92282261e-03],\n",
       "          [ 3.00414562e-02, -2.40332503e-02, -3.30817662e-02, ...,\n",
       "           -2.45024078e-02,  2.40731984e-03, -2.95338277e-02]],\n",
       " \n",
       "         [[-7.52781704e-03, -1.83135271e-03, -8.93036835e-03, ...,\n",
       "            8.75390694e-03, -3.09173763e-02,  2.15790607e-02],\n",
       "          [-9.81577486e-03, -1.94038302e-02,  1.93942115e-02, ...,\n",
       "           -2.59975828e-02, -2.83881593e-02,  2.28354037e-02],\n",
       "          [ 2.53983364e-02, -2.64297891e-02,  1.11409500e-02, ...,\n",
       "            7.12981448e-03, -2.45995633e-02, -2.34950595e-02],\n",
       "          ...,\n",
       "          [-3.11456621e-02, -8.12562555e-03, -3.20884921e-02, ...,\n",
       "            1.22451000e-02,  2.20740885e-02, -2.61719432e-02],\n",
       "          [ 2.55916230e-02, -2.52458081e-03, -3.20417061e-03, ...,\n",
       "           -1.67530254e-02, -2.83678994e-02, -2.73166317e-02],\n",
       "          [ 1.80033110e-02, -7.36334920e-03, -1.44934654e-03, ...,\n",
       "           -8.39347765e-03,  2.38229968e-02,  6.38496876e-03]],\n",
       " \n",
       "         [[ 2.45289654e-02, -3.16782147e-02, -1.79665387e-02, ...,\n",
       "            1.26809366e-02,  7.92670250e-03,  1.35359913e-02],\n",
       "          [-3.35652754e-03, -9.97150131e-03,  1.74021274e-02, ...,\n",
       "            3.35821882e-02, -1.41830798e-02,  6.35627657e-04],\n",
       "          [-1.27642937e-02,  2.18444616e-02, -2.54066456e-02, ...,\n",
       "           -1.97503846e-02, -1.94011983e-02,  2.71959752e-02],\n",
       "          ...,\n",
       "          [ 3.43789384e-02,  2.22311541e-03, -1.88217126e-02, ...,\n",
       "            3.22468579e-02, -3.11754644e-03, -3.60146351e-02],\n",
       "          [ 1.87532827e-02, -1.61783583e-02, -3.37985903e-03, ...,\n",
       "            2.97375172e-02, -2.40251627e-02, -8.15045461e-03],\n",
       "          [-2.23258957e-02, -3.07019874e-02,  8.17584991e-03, ...,\n",
       "           -1.04430690e-03, -4.99742851e-03, -1.97296217e-03]]],\n",
       " \n",
       " \n",
       "        [[[-3.77064198e-03, -1.40514076e-02,  3.12473997e-03, ...,\n",
       "           -1.70452502e-02,  4.95260581e-03,  1.62067562e-02],\n",
       "          [-3.20246741e-02, -2.82629468e-02, -5.92544675e-04, ...,\n",
       "            1.61682740e-02,  7.09541142e-04,  2.99089998e-02],\n",
       "          [-1.79542117e-02,  3.09822336e-03, -2.76656039e-02, ...,\n",
       "           -2.88782399e-02, -2.12765634e-02,  3.13446447e-02],\n",
       "          ...,\n",
       "          [ 3.55641544e-02, -1.04507077e-02,  2.57015303e-02, ...,\n",
       "            1.82767697e-02, -3.33344080e-02,  2.46540122e-02],\n",
       "          [ 2.27335058e-02, -2.62458362e-02,  1.23825669e-03, ...,\n",
       "           -2.32689679e-02,  1.74716078e-02, -1.06155779e-02],\n",
       "          [-3.28887179e-02, -1.96428355e-02,  1.97110698e-02, ...,\n",
       "            7.07823783e-03, -9.46028158e-03,  1.60475373e-02]],\n",
       " \n",
       "         [[ 1.49500370e-02,  3.29816416e-02,  6.11712411e-03, ...,\n",
       "            1.65541358e-02, -6.15977123e-03,  2.00371109e-02],\n",
       "          [ 2.31706947e-02, -6.18800707e-03,  2.52931789e-02, ...,\n",
       "            2.84038335e-02, -1.12612471e-02,  1.22675970e-02],\n",
       "          [-6.33169711e-03,  1.48254372e-02, -2.80504059e-02, ...,\n",
       "           -1.96444876e-02,  3.27354223e-02,  2.13306956e-02],\n",
       "          ...,\n",
       "          [ 2.13243365e-02, -1.34002939e-02,  1.96743757e-02, ...,\n",
       "           -3.58550027e-02,  3.57637107e-02,  7.77989626e-05],\n",
       "          [-9.87242907e-04,  2.67149508e-02, -2.01150309e-02, ...,\n",
       "            3.45423520e-02,  8.22580233e-03, -1.62243098e-03],\n",
       "          [ 3.20082009e-02, -3.50616351e-02,  1.38624720e-02, ...,\n",
       "           -2.18288042e-02,  8.88717175e-03,  3.42148542e-02]],\n",
       " \n",
       "         [[-8.21717083e-04, -7.70051777e-03,  2.09754519e-02, ...,\n",
       "            2.91871727e-02,  3.16305086e-02,  3.98155302e-04],\n",
       "          [ 1.97007880e-03, -1.01194847e-02, -1.08071379e-02, ...,\n",
       "           -8.76861066e-03,  2.39837170e-03, -3.45878415e-02],\n",
       "          [ 2.52396241e-02,  5.90135902e-03,  1.01979896e-02, ...,\n",
       "            2.40704864e-02,  2.00794563e-02, -3.57050449e-03],\n",
       "          ...,\n",
       "          [-3.43348719e-02, -6.84942864e-03,  5.12566045e-03, ...,\n",
       "           -1.77310035e-03,  6.05148450e-03, -2.76857950e-02],\n",
       "          [ 2.98322737e-02,  2.99353749e-02, -3.47863175e-02, ...,\n",
       "            1.25330128e-02,  8.11576843e-03,  2.43888982e-02],\n",
       "          [ 1.92771405e-02,  2.90701389e-02,  9.82903317e-03, ...,\n",
       "            1.43764988e-02,  4.22264263e-03,  1.36846527e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 2.67068520e-02, -8.08759220e-03, -1.26984715e-03, ...,\n",
       "           -2.35733408e-02, -3.06979008e-02,  4.32346389e-03],\n",
       "          [ 3.59680578e-02, -1.83928609e-02, -1.52207278e-02, ...,\n",
       "            1.27096102e-02,  3.58892232e-03,  3.45094725e-02],\n",
       "          [ 2.04045214e-02, -8.81859660e-03,  4.82223555e-03, ...,\n",
       "            4.37950343e-03, -9.38281417e-04, -3.07177734e-02],\n",
       "          ...,\n",
       "          [ 1.27420202e-03, -1.01196039e-02, -3.58694829e-02, ...,\n",
       "           -3.07212230e-02, -3.12504843e-02,  3.10293585e-02],\n",
       "          [ 3.14069912e-02,  1.26814283e-02,  3.42315286e-02, ...,\n",
       "           -3.60569470e-02, -8.91058147e-03,  3.51906419e-02],\n",
       "          [ 3.17445472e-02, -2.72636358e-02, -3.07547078e-02, ...,\n",
       "            3.31863910e-02, -3.03559229e-02, -1.51134040e-02]],\n",
       " \n",
       "         [[ 3.37650403e-02, -8.44929554e-03, -1.46205444e-02, ...,\n",
       "           -1.55002289e-02, -1.70308743e-02,  1.73619427e-02],\n",
       "          [-2.21195221e-02, -7.09670968e-03,  2.02417374e-03, ...,\n",
       "            2.20273435e-03, -3.57384980e-03,  1.46343783e-02],\n",
       "          [ 5.88979572e-03,  1.53369829e-02,  5.23240864e-03, ...,\n",
       "            3.35538387e-02, -7.62603059e-03, -2.11303364e-02],\n",
       "          ...,\n",
       "          [-2.73939669e-02, -2.78810002e-02,  2.34786123e-02, ...,\n",
       "           -2.92249732e-02, -5.59227914e-03, -2.01259553e-03],\n",
       "          [-2.56456602e-02,  1.02536269e-02,  1.47135966e-02, ...,\n",
       "            7.18577206e-03, -2.12485008e-02,  1.37853809e-02],\n",
       "          [-2.46965550e-02, -6.17685728e-03,  5.46171889e-03, ...,\n",
       "           -3.47919166e-02,  2.28288025e-02,  3.90019640e-03]],\n",
       " \n",
       "         [[ 1.50120743e-02, -4.95781191e-03,  3.53711620e-02, ...,\n",
       "            1.71365365e-02,  6.63549453e-03, -9.06862132e-03],\n",
       "          [-1.42283849e-02,  3.12267244e-03,  3.25668380e-02, ...,\n",
       "            1.33914500e-03, -1.96621846e-02, -2.68706679e-02],\n",
       "          [ 2.55449750e-02, -4.56926599e-03, -4.28268313e-03, ...,\n",
       "            5.85955381e-03, -1.66994780e-02, -2.90235821e-02],\n",
       "          ...,\n",
       "          [-2.89890226e-02,  2.90501565e-02,  2.30283290e-02, ...,\n",
       "           -1.54929925e-02,  1.13546848e-05, -9.45980102e-03],\n",
       "          [ 2.95802578e-02, -3.39899808e-02, -1.80229247e-02, ...,\n",
       "           -2.26369686e-02,  4.25285846e-03, -2.41981559e-02],\n",
       "          [-2.87987366e-02, -1.55718625e-03,  3.14870626e-02, ...,\n",
       "           -1.72373839e-02,  1.00427866e-02, -2.35540438e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1.], dtype=float32),\n",
       " array([[[[-0.04745565,  0.00034885,  0.06520121, ...,  0.03405457,\n",
       "           -0.0417626 , -0.05776646],\n",
       "          [-0.02407895, -0.06074461, -0.00714725, ...,  0.00120248,\n",
       "            0.00191573, -0.03401096],\n",
       "          [-0.04670536,  0.03233312,  0.00917152, ..., -0.01280204,\n",
       "           -0.0461383 ,  0.00248823],\n",
       "          ...,\n",
       "          [-0.01055065,  0.02536303,  0.05998819, ..., -0.01833301,\n",
       "           -0.00975397, -0.03230372],\n",
       "          [-0.06040781,  0.00968662,  0.03988951, ...,  0.06571215,\n",
       "           -0.06290015,  0.06007561],\n",
       "          [ 0.02314305, -0.02462834,  0.05637321, ..., -0.03103812,\n",
       "           -0.0113614 ,  0.02352585]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.05668654, -0.03993618,  0.03253502, ..., -0.05182405,\n",
       "           -0.03673138, -0.04017307],\n",
       "          [ 0.03901932, -0.02345218,  0.01926115, ...,  0.00748353,\n",
       "            0.00828786,  0.06167957],\n",
       "          [ 0.01357341,  0.01596962,  0.05194893, ...,  0.01604238,\n",
       "            0.05872317, -0.05901127],\n",
       "          ...,\n",
       "          [-0.01058395,  0.0527945 , -0.00620207, ...,  0.02426699,\n",
       "            0.01475181, -0.0240979 ],\n",
       "          [ 0.00889228,  0.03769225, -0.00095005, ..., -0.0113835 ,\n",
       "           -0.04875778, -0.00506471],\n",
       "          [ 0.04229163, -0.04922137, -0.05206993, ..., -0.03113127,\n",
       "           -0.01072383,  0.05475637]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[-1.68324094e-02,  1.18025653e-02,  8.41911510e-03, ...,\n",
       "            1.37305222e-02,  1.56132095e-02, -1.22091062e-02],\n",
       "          [-6.95113651e-03, -1.03895869e-02, -6.32947125e-03, ...,\n",
       "           -2.35569403e-02,  2.35878751e-02,  6.84124604e-03],\n",
       "          [-2.63127126e-03,  1.39888711e-02,  6.63633272e-03, ...,\n",
       "            1.68450326e-02, -1.80809796e-02, -2.17745062e-02],\n",
       "          ...,\n",
       "          [-2.22809650e-02,  2.42478475e-02, -2.12177243e-02, ...,\n",
       "           -5.40951267e-04, -2.10867506e-02,  1.80160254e-03],\n",
       "          [-2.19758283e-02,  2.34747492e-02, -1.45519106e-02, ...,\n",
       "           -1.40598752e-02,  5.88701293e-03, -1.14445295e-02],\n",
       "          [-1.25411060e-02,  2.99507007e-03,  2.37991028e-02, ...,\n",
       "            1.21470392e-02, -1.83443278e-02,  4.22108360e-03]],\n",
       " \n",
       "         [[-9.59382765e-03, -2.37732474e-02, -6.90136291e-03, ...,\n",
       "           -2.25422084e-02,  6.69663772e-04,  7.74697587e-03],\n",
       "          [ 1.40283927e-02, -2.29925364e-02,  1.77769810e-02, ...,\n",
       "           -2.18551233e-02, -2.52947472e-02, -1.63422711e-02],\n",
       "          [ 2.10927650e-02,  1.11193843e-02,  1.00219324e-02, ...,\n",
       "           -1.56849939e-02,  2.53445283e-02,  1.19315535e-02],\n",
       "          ...,\n",
       "          [ 1.43617988e-02, -1.76284499e-02, -1.24821039e-02, ...,\n",
       "           -1.68019924e-02,  1.32313371e-02, -1.11229746e-02],\n",
       "          [-2.26053223e-03,  7.79334456e-04,  1.13669895e-02, ...,\n",
       "           -3.31354141e-03,  1.15738139e-02, -8.26546177e-03],\n",
       "          [ 9.96352732e-03, -2.39862129e-04,  3.66615131e-03, ...,\n",
       "           -2.52151769e-02,  1.23375095e-03, -1.45310080e-02]],\n",
       " \n",
       "         [[-9.66586731e-03, -5.22824004e-04,  1.37433819e-02, ...,\n",
       "            6.74667954e-03,  5.83527610e-03,  2.21229419e-02],\n",
       "          [-2.11215597e-02,  1.39535032e-02,  2.68579088e-03, ...,\n",
       "            5.98490983e-03, -3.13226134e-04,  1.26332790e-03],\n",
       "          [ 2.17613652e-02,  1.83169544e-02, -1.73411630e-02, ...,\n",
       "           -1.94068328e-02,  5.99618256e-03,  1.21035092e-02],\n",
       "          ...,\n",
       "          [-2.06236467e-02, -1.46661317e-02, -9.26137157e-03, ...,\n",
       "           -2.15992853e-02, -2.33371481e-02,  1.61130689e-02],\n",
       "          [ 2.44588219e-03, -1.56679545e-02, -2.30375715e-02, ...,\n",
       "            2.02471055e-02,  9.57590714e-03,  1.76892579e-02],\n",
       "          [-2.00943574e-02,  1.65307149e-02,  1.01587176e-02, ...,\n",
       "            1.41399391e-02, -2.02865489e-02, -2.17394289e-02]]],\n",
       " \n",
       " \n",
       "        [[[-1.71726383e-03,  9.05125588e-03,  1.98060647e-02, ...,\n",
       "            2.37007588e-02,  1.52230412e-02, -1.74591355e-02],\n",
       "          [-1.94268040e-02, -9.85034555e-03, -1.93683673e-02, ...,\n",
       "           -2.07265839e-02, -1.84595287e-02,  2.41018534e-02],\n",
       "          [ 2.21279003e-02, -1.01711396e-02, -1.84163973e-02, ...,\n",
       "           -1.41716087e-02, -2.92885303e-03,  2.40483843e-02],\n",
       "          ...,\n",
       "          [-9.71691683e-04, -6.68130815e-05, -1.54018309e-02, ...,\n",
       "            1.46893784e-02, -9.23253037e-03, -2.40045376e-02],\n",
       "          [ 1.84510797e-02,  2.52922028e-02,  1.41429380e-02, ...,\n",
       "           -6.73364848e-05,  1.62814334e-02,  7.84726813e-03],\n",
       "          [-2.29007378e-03, -5.52266836e-03,  1.52891055e-02, ...,\n",
       "            1.51389204e-02,  1.53787062e-03,  2.52352878e-02]],\n",
       " \n",
       "         [[-1.15146590e-02,  2.08094344e-02,  3.17106210e-03, ...,\n",
       "            1.07655525e-02,  1.77418627e-02,  1.63038298e-02],\n",
       "          [-4.42444347e-03, -2.49570105e-02, -2.28546336e-02, ...,\n",
       "            1.56372525e-02,  2.32747085e-02,  7.13187456e-03],\n",
       "          [ 1.04948841e-02, -8.92934017e-03,  2.45748349e-02, ...,\n",
       "            2.51011550e-03,  2.31774300e-02, -1.08632566e-02],\n",
       "          ...,\n",
       "          [ 2.61745043e-03, -2.50155758e-02, -1.53083904e-02, ...,\n",
       "            1.61965080e-02, -2.17290260e-02, -1.95032414e-02],\n",
       "          [-5.90874441e-03, -1.88805349e-03, -2.30436306e-02, ...,\n",
       "            3.34555283e-04,  1.71774291e-02,  2.08751597e-02],\n",
       "          [ 1.67826973e-02, -2.05211900e-02,  2.25651562e-02, ...,\n",
       "           -1.33116813e-02,  6.05806708e-03, -1.74180437e-02]],\n",
       " \n",
       "         [[ 2.29708739e-02, -1.24387443e-04, -2.19410807e-02, ...,\n",
       "            6.83021918e-03,  1.38426088e-02,  1.23520643e-02],\n",
       "          [ 1.32224634e-02, -2.32719649e-02,  4.34022024e-03, ...,\n",
       "           -1.55339073e-02, -1.31772570e-02, -9.24515352e-03],\n",
       "          [ 1.45589076e-02, -5.87330200e-03,  1.06628351e-02, ...,\n",
       "           -9.35938582e-03,  6.83790073e-04,  2.13488713e-02],\n",
       "          ...,\n",
       "          [-1.68395825e-02, -4.40145470e-03,  2.03772448e-02, ...,\n",
       "           -2.46188976e-02,  2.04969309e-02,  3.08701955e-03],\n",
       "          [-1.09959962e-02, -1.68256089e-02,  7.03816116e-03, ...,\n",
       "           -2.05153506e-02,  1.48288012e-02, -2.24013180e-02],\n",
       "          [ 4.64328192e-03, -5.93813881e-03,  2.47353688e-02, ...,\n",
       "            1.36613287e-02, -6.01772778e-03,  1.84145048e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 2.32900679e-02, -1.77827105e-02,  1.82011314e-02, ...,\n",
       "            3.25242057e-04, -1.15237227e-02, -1.64891183e-02],\n",
       "          [-2.11212672e-02, -3.96096893e-03, -1.42199593e-02, ...,\n",
       "           -2.54676491e-02,  1.45503767e-02, -1.50638204e-02],\n",
       "          [ 9.71283019e-03, -4.33524884e-03, -8.63009319e-03, ...,\n",
       "           -5.74785098e-03,  5.98584488e-03,  1.41749047e-02],\n",
       "          ...,\n",
       "          [-1.64995268e-02, -1.37276947e-04, -8.93757679e-03, ...,\n",
       "           -1.74327288e-02,  2.01284364e-02, -2.52531003e-02],\n",
       "          [-2.39733830e-02, -1.83519535e-03, -1.04817189e-03, ...,\n",
       "            1.11883394e-02, -3.24243307e-03, -2.02030241e-02],\n",
       "          [-2.13121325e-02,  1.10499002e-03, -4.12860885e-03, ...,\n",
       "           -1.97915174e-03,  2.34855637e-02,  2.22277120e-02]],\n",
       " \n",
       "         [[-4.81654704e-03,  2.18138471e-02, -1.93209760e-03, ...,\n",
       "           -1.83538552e-02,  1.27718858e-02, -4.34111431e-03],\n",
       "          [-2.62649730e-04,  2.44172812e-02,  1.41986236e-02, ...,\n",
       "            6.54127076e-03, -2.23603584e-02,  1.77073330e-02],\n",
       "          [-2.12268010e-02, -4.88739461e-05, -1.39971431e-02, ...,\n",
       "           -6.05019554e-03, -1.53705748e-02,  2.19808407e-02],\n",
       "          ...,\n",
       "          [ 2.18421035e-03, -1.11278333e-03, -2.55109370e-03, ...,\n",
       "            4.17779386e-03,  8.75683501e-03,  7.04608113e-03],\n",
       "          [ 1.90025605e-02,  9.12055373e-03, -1.83149651e-02, ...,\n",
       "            2.23662779e-02, -4.16418537e-04,  2.13783681e-02],\n",
       "          [-1.67193934e-02,  1.76469348e-02, -4.83664125e-03, ...,\n",
       "            5.06452471e-03, -1.52436085e-02,  3.39544751e-03]],\n",
       " \n",
       "         [[ 4.91384417e-04, -1.97041761e-02,  1.04136243e-02, ...,\n",
       "            2.30279788e-02,  5.67524694e-03,  6.24120608e-03],\n",
       "          [-7.66050071e-03, -4.02957760e-03,  8.13316554e-04, ...,\n",
       "           -2.42412351e-02, -1.75729319e-02,  1.68571137e-02],\n",
       "          [ 2.26990134e-02, -3.48564610e-03,  2.28202939e-02, ...,\n",
       "            9.13996249e-04,  2.96310149e-03,  1.60053149e-02],\n",
       "          ...,\n",
       "          [-1.79756954e-02, -2.33324394e-02,  2.62980536e-03, ...,\n",
       "           -1.27639826e-02,  5.63620962e-03, -1.18138697e-02],\n",
       "          [-1.45440940e-02,  1.74313374e-02, -7.40526058e-03, ...,\n",
       "            1.82625018e-02, -9.03405435e-03, -2.48669274e-02],\n",
       "          [ 1.34582855e-02, -7.72623718e-03, -1.59491338e-02, ...,\n",
       "           -9.89687070e-03,  1.19577758e-02,  1.69289336e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[-0.02670315,  0.00526796,  0.00606367, ...,  0.00816883,\n",
       "            0.03908291, -0.03302778],\n",
       "          [-0.03461649,  0.02147733, -0.01746467, ..., -0.01558299,\n",
       "            0.03176   , -0.01579073],\n",
       "          [-0.04406864, -0.01765751, -0.01810074, ..., -0.01961329,\n",
       "            0.03254523,  0.02983948],\n",
       "          ...,\n",
       "          [-0.04309526,  0.01799412,  0.03035226, ..., -0.02060147,\n",
       "            0.01938212,  0.02945106],\n",
       "          [-0.02849926, -0.01208096,  0.04102805, ...,  0.01145338,\n",
       "           -0.01180408,  0.02831471],\n",
       "          [-0.01733053, -0.04149918, -0.03546492, ..., -0.00371052,\n",
       "           -0.03263929, -0.02778551]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([[[[-0.00257067,  0.02134053,  0.03903389, ...,  0.04181363,\n",
       "            0.024053  ,  0.01133088],\n",
       "          [-0.02447616,  0.01798388,  0.03206381, ..., -0.01847474,\n",
       "            0.03623492, -0.02848332],\n",
       "          [ 0.0148152 , -0.04702813, -0.0416748 , ..., -0.04395597,\n",
       "           -0.02859706,  0.02125265],\n",
       "          ...,\n",
       "          [ 0.02791376,  0.04553421,  0.0191888 , ...,  0.01405289,\n",
       "            0.0286967 ,  0.00251689],\n",
       "          [ 0.03463505,  0.03196531, -0.02607964, ...,  0.02871943,\n",
       "           -0.00138729, -0.04073707],\n",
       "          [-0.02943071, -0.00734159,  0.00937314, ..., -0.03551256,\n",
       "            0.04277278, -0.01317466]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.03224796,  0.02994145,  0.00938145, ..., -0.00321917,\n",
       "            0.02003131, -0.02573331],\n",
       "          [ 0.00232404, -0.01469978, -0.03158521, ..., -0.00939657,\n",
       "           -0.00237733, -0.01452073],\n",
       "          [-0.02877227,  0.00015212, -0.02884733, ..., -0.00434645,\n",
       "           -0.00816398, -0.03114735],\n",
       "          ...,\n",
       "          [ 0.0078993 ,  0.04498854,  0.01730159, ...,  0.04505218,\n",
       "           -0.02170012, -0.01287838],\n",
       "          [ 0.02515191,  0.04405077, -0.01252131, ...,  0.03707754,\n",
       "           -0.02705759, -0.02976533],\n",
       "          [-0.0020821 ,  0.03013747,  0.01941803, ..., -0.01000602,\n",
       "           -0.01904039, -0.02615466]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[ 0.02212285, -0.00602499,  0.01069392, ...,  0.0093701 ,\n",
       "           -0.02441189, -0.0175568 ],\n",
       "          [ 0.01037594, -0.01899594, -0.00316677, ..., -0.01078606,\n",
       "           -0.01145994, -0.01081887],\n",
       "          [-0.01456162,  0.00929568, -0.00341799, ..., -0.02247328,\n",
       "            0.01039496,  0.00513343],\n",
       "          ...,\n",
       "          [ 0.01820811, -0.01827654, -0.01238417, ...,  0.02377724,\n",
       "            0.01565   ,  0.02106116],\n",
       "          [-0.0142743 ,  0.01046424, -0.00980642, ...,  0.01965006,\n",
       "            0.01480327,  0.02151451],\n",
       "          [-0.01773609,  0.02264984,  0.01359963, ...,  0.0084173 ,\n",
       "            0.02484154, -0.00698273]],\n",
       " \n",
       "         [[-0.0043231 ,  0.00871749, -0.02545588, ...,  0.02548022,\n",
       "            0.01980895, -0.01818601],\n",
       "          [-0.02324511,  0.0032423 , -0.02391697, ..., -0.01978108,\n",
       "            0.02326861,  0.00118533],\n",
       "          [-0.01800139, -0.0025186 ,  0.00414229, ..., -0.02235502,\n",
       "            0.02048425, -0.00268013],\n",
       "          ...,\n",
       "          [-0.00567807, -0.00162582,  0.01249149, ..., -0.02333526,\n",
       "           -0.01711762,  0.02059337],\n",
       "          [-0.0111402 , -0.0004598 , -0.02314855, ..., -0.01361631,\n",
       "           -0.00829753,  0.01963158],\n",
       "          [-0.00836773,  0.00509201, -0.01311988, ...,  0.01513293,\n",
       "            0.00289458, -0.01261509]],\n",
       " \n",
       "         [[ 0.01456444,  0.02289307,  0.00952743, ...,  0.00189079,\n",
       "            0.00689352,  0.00792554],\n",
       "          [-0.02009132, -0.01710512, -0.02045457, ..., -0.00819353,\n",
       "            0.02182356, -0.01055165],\n",
       "          [ 0.02454765, -0.0118834 ,  0.0110977 , ..., -0.01317111,\n",
       "            0.01689871, -0.02421956],\n",
       "          ...,\n",
       "          [-0.01002342,  0.00100261, -0.00306605, ...,  0.02344961,\n",
       "            0.01784207, -0.00019473],\n",
       "          [ 0.01074829, -0.00838684, -0.00998424, ..., -0.02443309,\n",
       "           -0.00181242, -0.006565  ],\n",
       "          [ 0.01392163,  0.02488961, -0.0249799 , ..., -0.00109348,\n",
       "            0.01842947, -0.02490113]]],\n",
       " \n",
       " \n",
       "        [[[-0.01050333, -0.00488104,  0.01701603, ..., -0.0016829 ,\n",
       "            0.00848407, -0.02057507],\n",
       "          [ 0.00090812,  0.00177899, -0.022035  , ...,  0.0213956 ,\n",
       "            0.00755442, -0.01969427],\n",
       "          [ 0.00233348, -0.02444741,  0.01716702, ..., -0.02481589,\n",
       "            0.02252226,  0.01018567],\n",
       "          ...,\n",
       "          [ 0.01032097,  0.02406819,  0.02538365, ...,  0.02513604,\n",
       "           -0.02482043, -0.00939957],\n",
       "          [-0.01322867,  0.01618147,  0.01838141, ..., -0.01128619,\n",
       "           -0.01494873, -0.00916864],\n",
       "          [-0.00936557, -0.01828584, -0.00721942, ...,  0.00327932,\n",
       "           -0.01976474, -0.00296276]],\n",
       " \n",
       "         [[-0.01455776, -0.01551231, -0.00427289, ..., -0.01921709,\n",
       "           -0.00803114, -0.01900081],\n",
       "          [ 0.01652672,  0.02307819,  0.00871473, ...,  0.00745707,\n",
       "           -0.01159905,  0.01675811],\n",
       "          [ 0.02342462,  0.00318679, -0.00948405, ...,  0.0118901 ,\n",
       "           -0.00372421,  0.02206196],\n",
       "          ...,\n",
       "          [-0.00364002,  0.00405411,  0.00642117, ..., -0.01242681,\n",
       "           -0.00157756, -0.00496608],\n",
       "          [-0.00279883,  0.0020544 , -0.01661655, ...,  0.0044814 ,\n",
       "            0.01658624,  0.00015006],\n",
       "          [-0.02256345, -0.01441842, -0.001337  , ..., -0.01518179,\n",
       "           -0.02118868, -0.02352495]],\n",
       " \n",
       "         [[ 0.02275923, -0.02291944,  0.02107972, ..., -0.0118124 ,\n",
       "            0.00401537, -0.01046918],\n",
       "          [-0.01931423, -0.0219032 , -0.01438014, ...,  0.01805228,\n",
       "            0.00274182, -0.0168188 ],\n",
       "          [ 0.01780508, -0.00667878, -0.01556661, ...,  0.02226995,\n",
       "            0.00328093, -0.00274265],\n",
       "          ...,\n",
       "          [-0.00122278, -0.01198574, -0.00719191, ...,  0.02019419,\n",
       "            0.00252441,  0.01868235],\n",
       "          [-0.02190026,  0.0058798 ,  0.01501483, ..., -0.01105524,\n",
       "           -0.00863587,  0.01684574],\n",
       "          [-0.01262061,  0.01848413, -0.01820099, ..., -0.02282082,\n",
       "           -0.01386119,  0.01793277]]],\n",
       " \n",
       " \n",
       "        [[[-0.01482482, -0.02209407,  0.0037843 , ..., -0.01491688,\n",
       "            0.00942941,  0.00136044],\n",
       "          [ 0.00984312,  0.01134595, -0.01514304, ...,  0.01650197,\n",
       "            0.00051207,  0.00178153],\n",
       "          [-0.01388952, -0.01779874, -0.00066434, ..., -0.0191762 ,\n",
       "           -0.01506436,  0.02302428],\n",
       "          ...,\n",
       "          [ 0.01869148, -0.02505119, -0.00168946, ..., -0.02032579,\n",
       "            0.01109225,  0.02136914],\n",
       "          [ 0.01541527,  0.02359831, -0.02537488, ...,  0.0101945 ,\n",
       "            0.00011935,  0.0047249 ],\n",
       "          [ 0.0015591 , -0.00081883,  0.02434627, ...,  0.0238648 ,\n",
       "            0.00220924, -0.01992617]],\n",
       " \n",
       "         [[ 0.00496724, -0.01251152,  0.00291981, ...,  0.01904909,\n",
       "            0.02237892,  0.01413507],\n",
       "          [-0.00673606,  0.01886524,  0.00446809, ...,  0.00309292,\n",
       "            0.02141734, -0.02387192],\n",
       "          [-0.00381479,  0.01350024, -0.00138059, ...,  0.01226813,\n",
       "            0.00856612, -0.01910255],\n",
       "          ...,\n",
       "          [ 0.00464826, -0.0244541 , -0.0033649 , ..., -0.01354294,\n",
       "           -0.0068635 ,  0.02002246],\n",
       "          [-0.00872416,  0.01593967,  0.01324526, ...,  0.02434411,\n",
       "            0.0029409 , -0.01096793],\n",
       "          [ 0.01807358,  0.00852232,  0.01742145, ..., -0.00301151,\n",
       "            0.02363194,  0.00739402]],\n",
       " \n",
       "         [[ 0.01523803, -0.0050589 , -0.01905106, ...,  0.00093804,\n",
       "            0.00446943,  0.01666072],\n",
       "          [ 0.00501025, -0.01843151,  0.00321924, ...,  0.01159213,\n",
       "           -0.01766949, -0.02115808],\n",
       "          [ 0.02354903,  0.0028527 ,  0.01035494, ..., -0.01798394,\n",
       "           -0.01924062, -0.01582141],\n",
       "          ...,\n",
       "          [ 0.00640684,  0.02028194, -0.01946877, ...,  0.00439257,\n",
       "            0.02520273,  0.02136565],\n",
       "          [ 0.02072236,  0.00569138, -0.01289819, ..., -0.00500104,\n",
       "           -0.00835016,  0.02243453],\n",
       "          [-0.00824557,  0.00674343, -0.01777573, ..., -0.01827763,\n",
       "            0.0034112 , -0.02084118]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[-9.8337010e-03, -4.5478493e-02,  5.5609271e-03, ...,\n",
       "            3.4448899e-02,  1.2374908e-02, -3.3421244e-02],\n",
       "          [-1.3378587e-02,  5.2771345e-03,  1.1048757e-02, ...,\n",
       "            1.5915744e-02, -6.3584484e-03, -3.3090912e-02],\n",
       "          [-4.7150940e-02,  2.4992749e-02, -2.9933805e-02, ...,\n",
       "           -1.0576975e-02, -1.0268595e-02, -9.3913041e-03],\n",
       "          ...,\n",
       "          [ 2.6963018e-03, -6.5335631e-04,  3.3139832e-02, ...,\n",
       "           -3.3215478e-02,  3.0600958e-02,  4.6633832e-02],\n",
       "          [-3.8641579e-02, -2.4260592e-02,  3.5148069e-02, ...,\n",
       "            7.8822412e-03,  3.5620049e-02, -1.8351512e-02],\n",
       "          [-3.7282705e-05, -2.3261413e-02,  4.1673653e-02, ...,\n",
       "           -4.1549291e-02, -4.7349863e-02, -3.6470480e-03]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[[[-0.01190788, -0.01113559,  0.01808088, ..., -0.02152757,\n",
       "           -0.03641799, -0.03309313],\n",
       "          [ 0.04361451,  0.01084329, -0.02209269, ...,  0.02198917,\n",
       "            0.00264144, -0.03365762],\n",
       "          [-0.02648561, -0.01905785,  0.00306854, ..., -0.03602294,\n",
       "            0.02469889,  0.04064566],\n",
       "          ...,\n",
       "          [ 0.03893967, -0.03256363, -0.02423626, ...,  0.03082558,\n",
       "           -0.03313769, -0.03078555],\n",
       "          [ 0.04674102, -0.02213114, -0.02961566, ...,  0.04018103,\n",
       "            0.03762302,  0.03397349],\n",
       "          [ 0.03282214,  0.0280823 ,  0.01494402, ...,  0.04768107,\n",
       "            0.0318499 , -0.03720862]]]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[ 1.81533657e-02, -1.61947627e-02,  1.14689842e-02, ...,\n",
       "            4.32045385e-03, -1.91068053e-02,  7.00752810e-03],\n",
       "          [-1.65861659e-02, -2.07243264e-02, -1.32678198e-02, ...,\n",
       "            2.24030539e-02,  2.13943124e-02, -2.46800892e-02],\n",
       "          [-1.75309032e-02, -1.58548597e-02, -2.26457901e-02, ...,\n",
       "            8.64202157e-03, -2.22431142e-02,  7.45320320e-03],\n",
       "          ...,\n",
       "          [-2.18521245e-02, -1.47960000e-02, -6.85210526e-03, ...,\n",
       "            1.51176229e-02, -1.32568944e-02,  1.48340836e-02],\n",
       "          [-1.81913935e-02, -2.15097144e-02,  1.56575814e-02, ...,\n",
       "           -1.19187767e-02,  5.74601442e-03, -2.02245452e-03],\n",
       "          [ 2.37279683e-02, -4.03087959e-03,  1.47693492e-02, ...,\n",
       "           -1.10871494e-02, -2.18715407e-02,  8.91872495e-03]],\n",
       " \n",
       "         [[-1.90580450e-03, -1.53544722e-02, -2.41065118e-02, ...,\n",
       "            7.58100301e-03,  2.25683786e-02,  1.52812824e-02],\n",
       "          [ 1.62432529e-02, -1.69016197e-02, -9.34778526e-03, ...,\n",
       "           -1.99997798e-02,  9.05606896e-04, -1.48781259e-02],\n",
       "          [ 2.62862071e-04,  6.77921623e-03, -1.28054991e-04, ...,\n",
       "           -8.61610658e-03,  1.46359950e-02, -6.23008050e-03],\n",
       "          ...,\n",
       "          [ 3.24309617e-03,  1.12638772e-02,  1.43177174e-02, ...,\n",
       "            1.94824301e-03,  8.60583782e-03,  4.85785306e-03],\n",
       "          [ 1.67006925e-02, -4.01521102e-04, -1.65730510e-02, ...,\n",
       "           -1.64039880e-02, -4.19669412e-03, -1.58006269e-02],\n",
       "          [ 3.85291688e-03,  4.98552620e-03,  1.81336366e-02, ...,\n",
       "            2.80338898e-03, -1.06454082e-03,  1.79418512e-02]],\n",
       " \n",
       "         [[ 2.38665044e-02,  1.80647485e-02, -1.69922132e-02, ...,\n",
       "            9.66760516e-03, -2.03980766e-02, -6.42793253e-04],\n",
       "          [-2.46237032e-02, -7.64069892e-03, -6.39233552e-03, ...,\n",
       "           -1.18115880e-02,  3.56303900e-03,  2.21152045e-02],\n",
       "          [-1.13643873e-02,  8.78508389e-03, -1.08190915e-02, ...,\n",
       "           -1.87242106e-02,  1.05818659e-02, -7.14526325e-03],\n",
       "          ...,\n",
       "          [-1.81371290e-02,  5.03939390e-03,  2.56661139e-03, ...,\n",
       "           -4.28469852e-04, -6.68752007e-03,  1.29350536e-02],\n",
       "          [ 1.48755386e-02,  6.43710420e-03,  2.35889107e-04, ...,\n",
       "           -2.26902552e-02,  5.70950843e-03,  1.60474889e-02],\n",
       "          [ 8.06052610e-03,  1.56905428e-02,  7.59829208e-03, ...,\n",
       "            7.27929920e-03, -3.53566930e-03,  5.04342094e-03]]],\n",
       " \n",
       " \n",
       "        [[[ 1.34005882e-02, -3.63420136e-03, -1.27609167e-02, ...,\n",
       "           -1.70897059e-02, -1.40996305e-02,  1.62865557e-02],\n",
       "          [ 1.54317692e-02, -8.85370001e-03,  3.42902169e-03, ...,\n",
       "           -1.84614696e-02,  2.17403471e-02,  1.29468180e-03],\n",
       "          [ 8.22856650e-03,  5.32800704e-03, -1.01049710e-02, ...,\n",
       "           -2.17619799e-02,  2.37295516e-02, -9.95393842e-03],\n",
       "          ...,\n",
       "          [-1.91828795e-02, -2.32343581e-02, -2.49401238e-02, ...,\n",
       "           -8.60469416e-03, -2.23609488e-02,  1.91147141e-02],\n",
       "          [ 6.83432445e-03, -2.53243558e-03, -1.36122108e-03, ...,\n",
       "           -2.43472625e-02,  2.15329714e-02, -9.62364860e-03],\n",
       "          [-2.44066175e-02, -8.40441138e-03,  8.29115137e-03, ...,\n",
       "           -2.14515813e-02,  1.19381957e-02,  1.94621086e-03]],\n",
       " \n",
       "         [[ 1.21940374e-02,  1.97703876e-02, -1.94312260e-02, ...,\n",
       "           -2.00476907e-02,  1.30453892e-02, -2.53294110e-02],\n",
       "          [ 2.54632942e-02,  1.76052004e-02,  1.98813342e-02, ...,\n",
       "            2.68601626e-03,  1.58574469e-02, -1.39377089e-02],\n",
       "          [-1.54191200e-02,  1.18114054e-02, -7.94309750e-03, ...,\n",
       "            8.95750150e-03,  1.06386878e-02, -1.79188512e-02],\n",
       "          ...,\n",
       "          [-1.70123614e-02, -3.93266417e-03,  1.91413052e-02, ...,\n",
       "           -1.51837729e-02,  2.41977498e-02,  1.51075125e-02],\n",
       "          [-1.08091999e-02, -1.58518180e-02, -1.56598091e-02, ...,\n",
       "           -1.53260874e-02, -7.69499876e-03, -1.31714232e-02],\n",
       "          [-2.45431252e-03, -9.30028036e-03, -1.46434084e-03, ...,\n",
       "           -4.59809974e-03, -2.52256524e-02, -8.26541334e-03]],\n",
       " \n",
       "         [[-2.10786536e-02, -2.35088635e-02,  2.09816732e-02, ...,\n",
       "           -1.48524716e-02,  1.08363666e-03, -1.09317377e-02],\n",
       "          [ 2.18805075e-02, -2.43613198e-02, -2.12524794e-02, ...,\n",
       "            9.06590000e-03, -1.20881628e-02, -2.27232389e-02],\n",
       "          [-1.64653882e-02,  9.43855569e-03, -2.08487678e-02, ...,\n",
       "            2.49706209e-02,  2.76305713e-03, -7.36166723e-03],\n",
       "          ...,\n",
       "          [ 1.96820088e-02,  1.43321715e-02,  1.07156746e-02, ...,\n",
       "           -2.37493571e-02, -4.79055941e-03,  1.74455419e-02],\n",
       "          [ 1.99936181e-02,  6.80837035e-03,  1.76949948e-02, ...,\n",
       "            1.54768340e-02, -1.80198234e-02, -2.42741220e-03],\n",
       "          [ 1.46452188e-02, -1.23083750e-02,  5.26224077e-03, ...,\n",
       "           -3.56019661e-03, -4.68480028e-03, -1.63554121e-02]]],\n",
       " \n",
       " \n",
       "        [[[ 1.61723457e-02, -2.02672519e-02,  5.29628247e-03, ...,\n",
       "           -4.17395495e-03, -1.84963085e-02, -1.98011324e-02],\n",
       "          [-7.25016557e-03, -1.96230840e-02, -2.00862717e-02, ...,\n",
       "            1.37577988e-02,  1.08839944e-02, -7.01254793e-03],\n",
       "          [-1.70470793e-02, -1.63494982e-02,  1.21153370e-02, ...,\n",
       "            2.52405554e-02,  3.96965072e-03,  2.49556527e-02],\n",
       "          ...,\n",
       "          [ 2.19237432e-03,  1.13633424e-02, -1.38870403e-02, ...,\n",
       "            1.35957561e-02, -8.20159167e-03, -2.29181554e-02],\n",
       "          [ 1.87421925e-02, -1.12445503e-02, -6.87567331e-03, ...,\n",
       "           -1.72112137e-02, -1.53176300e-03, -9.57426243e-03],\n",
       "          [-3.45795415e-03,  1.14009008e-02, -5.89684583e-03, ...,\n",
       "            1.41559131e-02, -1.21022947e-02, -1.66088026e-02]],\n",
       " \n",
       "         [[-1.71444137e-02,  9.32347029e-05,  4.32720780e-03, ...,\n",
       "           -1.71289183e-02,  2.60160305e-03,  7.30887055e-04],\n",
       "          [-2.22379379e-02,  2.14472152e-02, -1.69892330e-02, ...,\n",
       "            2.08476856e-02,  2.37474032e-02, -1.21300407e-02],\n",
       "          [ 9.13252681e-03, -7.43537396e-03, -6.43717684e-03, ...,\n",
       "            6.31569698e-03,  1.57676600e-02,  2.48469152e-02],\n",
       "          ...,\n",
       "          [ 1.91276371e-02,  1.73259899e-03, -5.22615388e-03, ...,\n",
       "            6.34546205e-03, -2.45850123e-02,  2.41382420e-02],\n",
       "          [-1.79235712e-02,  4.40848805e-03, -3.99523228e-03, ...,\n",
       "            6.54564798e-03,  1.45068318e-02, -5.36367297e-05],\n",
       "          [ 1.09988190e-02, -2.25889422e-02,  1.36118121e-02, ...,\n",
       "            8.73205066e-03,  1.09890476e-02,  1.37210861e-02]],\n",
       " \n",
       "         [[-1.50591610e-02, -2.41007991e-02,  1.27005652e-02, ...,\n",
       "           -1.47071462e-02, -1.29436515e-03,  2.15494558e-02],\n",
       "          [-3.81403230e-03, -1.03775598e-03,  1.89559981e-02, ...,\n",
       "           -1.71925817e-02,  1.48289129e-02, -2.04096828e-02],\n",
       "          [-7.24331476e-03,  8.29797611e-03, -1.10046286e-02, ...,\n",
       "           -1.34813711e-02, -5.87597303e-03,  1.94011256e-02],\n",
       "          ...,\n",
       "          [-4.07471135e-03, -1.25834644e-02,  1.61037482e-02, ...,\n",
       "            1.21279657e-02, -1.42431548e-02,  1.82572082e-02],\n",
       "          [-1.81054529e-02, -2.24734005e-02, -4.55790013e-03, ...,\n",
       "            1.55441463e-04, -2.77418830e-03, -3.16538662e-03],\n",
       "          [-8.76077078e-03, -1.70560833e-02,  1.28861256e-02, ...,\n",
       "            1.81770734e-02, -8.04400817e-03,  1.42482407e-02]]]],\n",
       "       dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
       "        1., 1.], dtype=float32),\n",
       " array([[[[ 0.02305606, -0.01957658,  0.0345671 , ...,  0.01263732,\n",
       "            0.02034794,  0.02073599],\n",
       "          [ 0.02830745,  0.01489872, -0.02046233, ...,  0.01857751,\n",
       "            0.00928999, -0.04138402],\n",
       "          [-0.01334337, -0.03808062, -0.02187867, ..., -0.02846742,\n",
       "           -0.02524295, -0.03075906],\n",
       "          ...,\n",
       "          [ 0.01354598,  0.03558149, -0.03756461, ...,  0.03867475,\n",
       "            0.00979049,  0.0434269 ],\n",
       "          [-0.04832162,  0.04047123,  0.00920289, ..., -0.00987472,\n",
       "           -0.03919793, -0.01778524],\n",
       "          [-0.02948799, -0.03083151,  0.00493534, ..., -0.02671115,\n",
       "           -0.01104414, -0.00245104]]]], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([0., 0., 0., ..., 0., 0., 0.], dtype=float32),\n",
       " array([1., 1., 1., ..., 1., 1., 1.], dtype=float32),\n",
       " array([[ 0.02746863, -0.04251747,  0.04249675, ...,  0.0342112 ,\n",
       "          0.01298128,  0.00806882],\n",
       "        [ 0.03729189,  0.00874088, -0.03378506, ..., -0.03643234,\n",
       "         -0.02693742, -0.03936215],\n",
       "        [ 0.00621432,  0.01291423, -0.03626003, ..., -0.02583247,\n",
       "         -0.01762375,  0.02813973],\n",
       "        ...,\n",
       "        [-0.01825251, -0.02315427, -0.03632054, ...,  0.02486241,\n",
       "          0.03650174, -0.04456953],\n",
       "        [ 0.03970531,  0.03996421,  0.01095073, ..., -0.01721127,\n",
       "          0.03797176, -0.01904144],\n",
       "        [ 0.00351903, -0.01161126, -0.02743796, ..., -0.02628196,\n",
       "         -0.04150292, -0.014393  ]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
       "        0., 0.], dtype=float32),\n",
       " array([[-0.07113522, -0.00935549,  0.07525996,  0.03196157, -0.01048312,\n",
       "         -0.018814  ],\n",
       "        [-0.00253433,  0.05156372,  0.03209414,  0.07787458, -0.02864005,\n",
       "         -0.04298275],\n",
       "        [ 0.07679562,  0.03169776, -0.0848141 , -0.10733899, -0.02673347,\n",
       "         -0.02565894],\n",
       "        ...,\n",
       "        [ 0.06629937, -0.01896182, -0.08588783, -0.021902  , -0.08533232,\n",
       "         -0.05982041],\n",
       "        [ 0.07577097, -0.02087572,  0.07162306, -0.0330891 ,  0.03402802,\n",
       "         -0.08059286],\n",
       "        [-0.08613382, -0.04989215,  0.09653854,  0.08682392, -0.08482878,\n",
       "         -0.01950924]], dtype=float32),\n",
       " array([0., 0., 0., 0., 0., 0.], dtype=float32)]"
      ]
     },
     "execution_count": 76,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "GF_model.get_weights()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 101,
   "metadata": {},
   "outputs": [],
   "source": [
    "fault_data = [87, 8]\n",
    "import math\n",
    "STD = 0.1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 102,
   "metadata": {},
   "outputs": [],
   "source": [
    "GF_model = clone_model(model)\n",
    "GF_model.set_weights(model.get_weights())\n",
    "layers = GF_model.layers\n",
    "# layer_utils = LayerUtils()\n",
    "data = fault_data\n",
    "# for  in fault_data[:10]:\n",
    "layer = layers[data[0]]\n",
    "# if not layer_utils.is_layer_in_weight_change_white_list(layer):\n",
    "#     continue\n",
    "\n",
    "weights = layer.get_weights()\n",
    "if len(weights) > 0:\n",
    "    # assert len(weights) == 2\n",
    "    if len(weights) == 1:\n",
    "        weights_w = weights[0]\n",
    "        weights_w = weights_w.transpose()\n",
    "        STD = math.sqrt(weights_w[data[1]].var()) * STD\n",
    "        weights_w[data[1]] += np.random.normal(scale=STD, size=weights_w[data[1]].shape)\n",
    "        weights_w = weights_w.transpose()\n",
    "        weights = [weights_w]\n",
    "        layer.set_weights(weights)\n",
    "    else:\n",
    "        weights_w, weights_b = weights\n",
    "        weights_w = weights_w.transpose()\n",
    "        STD = math.sqrt(weights_w[data[1]].var()) * STD\n",
    "        weights_w[data[1]] += np.random.normal(scale=STD, size=weights_w[data[1]].shape)\n",
    "        weights_w = weights_w.transpose()\n",
    "        weights = weights_w, weights_b\n",
    "        layer.set_weights(weights)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 103,
   "metadata": {},
   "outputs": [],
   "source": [
    "GF_model.save_weights('model3.h5')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "model = load_model('./data/neural_networks/BIT_RESNET50_model.h5')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "X_data_perdict = model(X_data)\n",
    "X_data_perdict_arg = np.argmax(X_data_perdict, axis=1)\n",
    "Y_data_arg = np.argmax(Y_data, axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "json_config = model.to_json()\n",
    "json_config"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "with open('Resnet_model.json', 'w') as json_file:\n",
    "    json_file.write(json_config)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "grad",
   "language": "python",
   "name": "grad"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.15"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
